如何在内核中获得设备架构？

Question

正如标题所示。两者都有

let blockReducer = BlockReduce.RakingCommutativeOnly<float32>(dims,DeviceArch.Create("sm35"))

和

let blockReducer = BlockReduce.RakingCommutativeOnly<float32>(dims,worker.Device.Arch)

编译失败。

Answer 1

实际上，let blockReducer = BlockReduce.RakingCommutativeOnly<float32>(dims,worker.Device.Arch)应该在内核外部调用。那么就没有任何问题了。