求助 Type f64 not supported.

在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 后端。