parent
3a4e04ab1a
commit
94e8f70dd4
@ -0,0 +1,35 @@
|
|||||||
|
//
|
||||||
|
// Capabilities.swift
|
||||||
|
// Diffusion-macOS
|
||||||
|
//
|
||||||
|
// Created by Pedro Cuenca on 20/2/23.
|
||||||
|
//
|
||||||
|
|
||||||
|
import Foundation
|
||||||
|
|
||||||
|
let runningOnMac = true
|
||||||
|
|
||||||
|
#if canImport(MLCompute)
|
||||||
|
import MLCompute
|
||||||
|
let _hasANE = MLCDevice.ane() != nil
|
||||||
|
#else
|
||||||
|
let _hasANE = false
|
||||||
|
#endif
|
||||||
|
|
||||||
|
final class Capabilities {
|
||||||
|
static let hasANE = _hasANE
|
||||||
|
|
||||||
|
// According to my tests this is a good proxy to estimate whether CPU+GPU
|
||||||
|
// or CPU+NE works better. Things may become more complicated if we
|
||||||
|
// choose all compute units.
|
||||||
|
static var performanceCores: Int = {
|
||||||
|
var ncores: Int32 = 0
|
||||||
|
var bytes = MemoryLayout<Int32>.size
|
||||||
|
|
||||||
|
// In M1/M2 perflevel0 refers to the performance cores and perflevel1 are the efficiency cores
|
||||||
|
// In Intel there's only one performance level
|
||||||
|
let result = sysctlbyname("hw.perflevel0.physicalcpu", &ncores, &bytes, nil, 0)
|
||||||
|
guard result == 0 else { return 0 }
|
||||||
|
return Int(ncores)
|
||||||
|
}()
|
||||||
|
}
|
Loading…
Reference in new issue