diff --git a/build2cmake/src/torch/metal.rs b/build2cmake/src/torch/metal.rs index aa96140..244a9a9 100644 --- a/build2cmake/src/torch/metal.rs +++ b/build2cmake/src/torch/metal.rs @@ -30,7 +30,7 @@ pub fn write_torch_ext_metal( let mut file_set = FileSet::default(); - let ops_name = kernel_ops_identifier(&target_dir, &build.general.name, ops_id); + let ops_name = kernel_ops_identifier(&target_dir, &build.general.python_name(), ops_id); write_cmake( env,