Apple平台如何无需通过设备型号版本判断是否搭载Neural Engine

结论 #

Apple 没有提供直接查询设备是否搭载 Neural Engine 的公开 API，但你可以通过 CoreML 的官方能力间接检测，完全不需要依赖设备型号、系统版本的匹配规则，适配你的使用场景。

实现方案 #

你可以通过「强制指定 Neural Engine 运行测试模型」的方式做精准判断：

• 提前准备一个极小的、仅包含基础运算的 CoreML 测试模型（仅需几KB大小，可通过 Core ML Tools 快速生成，无需训练）
• 加载模型时强制指定仅使用 Neural Engine 计算单元，能成功完成推理就说明设备具备可用的 Neural Engine 硬件加速能力，反之则没有
  对应 Swift 参考代码如下：

import CoreML

func isNeuralEngineAvailable() -> Bool {
    let config = MLModelConfiguration()
    // 强制指定仅允许使用 Neural Engine 运行
    config.computeUnits = .neuralEngine
    
    do {
        // NeuralEngineTestModel 为你提前打包到App的测试模型
        let testModel = try NeuralEngineTestModel(configuration: config)
        // 执行一次简单推理验证实际可用
        let testInput = NeuralEngineTestModelInput(feature: MLMultiArray(shape: [1, 1], dataType: .float32)!)
        let _ = try testModel.prediction(input: testInput)
        return true
    } catch {
        return false
    }
}

注意事项 #

• 该方案使用的全部是公开API，符合App Store审核要求
• 检测结果会自动覆盖所有边缘场景：包括旧设备无硬件、系统配置禁用 Neural Engine、特殊定制设备无对应模块等，无需手动维护设备型号匹配表
• 检测耗时极低，通常在1ms以内，不会影响App启动或运行性能
• 如果你不需要做非常精准的检测，也可以在iOS14+/macOS11+以上系统通过MLDevice的supportsNeuralEngine属性做快速判断，但该属性仅能判断设备硬件支持，无法校验当前App是否有权限使用Neural Engine

内容的提问来源于stack exchange，提问作者Rеnаud