在M2 Max上运行x.from_numpy(np.zeros([H, W]))
时出现以下报错:
RuntimeError: [spirv_ir_builder.cpp:get_primitive_type@296] Type f64 not supported.
在CUDA上相同代码无报错,求助大家这可能是什么问题?
在M2 Max上运行x.from_numpy(np.zeros([H, W]))
时出现以下报错:
RuntimeError: [spirv_ir_builder.cpp:get_primitive_type@296] Type f64 not supported.
在CUDA上相同代码无报错,求助大家这可能是什么问题?
metal 只支持 fp32 的,一定需要 fp64 的话可以用 cpu 后端。