fix pip3 install error for example like cutlass-gemm in local build.

build2cmake/src/templates/cpu/setup.py

@@ -104,16 +104,17 @@ def build_extension(self, ext: CMakeExtension) -> None:
         )
 
 
+{% set python_name = name | replace('-', '_') %}
 setup(
-    name="{{ name }}",
+    name="{{ python_name }}",
     # The version is just a stub, it's not used by the final build artefact.
     version="0.1.0",
-    ext_modules=[CMakeExtension("{{ name }}.{{ ops_name }}")],
+    ext_modules=[CMakeExtension("{{ python_name }}.{{ ops_name }}")],
     cmdclass={"build_ext": CMakeBuild},
-    packages=find_packages(where="torch-ext", include=["{{ name }}*"]),
+    packages=find_packages(where="torch-ext", include=["{{ python_name }}*"]),
     package_dir={"": "torch-ext"},
 {% if data_globs %}
-    package_data={"{{ name }}": [ {{ data_globs }} ]},
+    package_data={"{{ python_name }}": [ {{ data_globs }} ]},
 {% endif %}
     zip_safe=False,
     install_requires=["torch"],

build2cmake/src/templates/cuda/setup.py

@@ -124,16 +124,17 @@ def build_extension(self, ext: CMakeExtension) -> None:
 
 
 
+{% set python_name = name | replace('-', '_') %}
 setup(
-    name="{{ name }}",
+    name="{{ python_name }}",
     # The version is just a stub, it's not used by the final build artefact.
     version="0.1.0",
-    ext_modules=[CMakeExtension("{{ name }}.{{ ops_name }}")],
+    ext_modules=[CMakeExtension("{{ python_name }}.{{ ops_name }}")],
     cmdclass={"build_ext": CMakeBuild},
-    packages=find_packages(where="torch-ext", include=["{{ name }}*"]),
+    packages=find_packages(where="torch-ext", include=["{{ python_name }}*"]),
     package_dir={"": "torch-ext"},
 {% if data_globs %}
-    package_data={"{{ name }}": [ {{ data_globs }} ]},
+    package_data={"{{ python_name }}": [ {{ data_globs }} ]},
 {% endif %}
     zip_safe=False,
     install_requires=["torch"],

build2cmake/src/templates/metal/setup.py

@@ -104,16 +104,17 @@ def build_extension(self, ext: CMakeExtension) -> None:
         )
 
 
+{% set python_name = name | replace('-', '_') %}
 setup(
-    name="{{ name }}",
+    name="{{ python_name }}",
     # The version is just a stub, it's not used by the final build artefact.
     version="0.1.0",
-    ext_modules=[CMakeExtension("{{ name }}.{{ ops_name }}")],
+    ext_modules=[CMakeExtension("{{ python_name }}.{{ ops_name }}")],
     cmdclass={"build_ext": CMakeBuild},
-    packages=find_packages(where="torch-ext", include=["{{ name }}*"]),
+    packages=find_packages(where="torch-ext", include=["{{ python_name }}*"]),
     package_dir={"": "torch-ext"},
 {% if data_globs %}
-    package_data={"{{ name }}": [ {{ data_globs }} ]},
+    package_data={"{{ python_name }}": [ {{ data_globs }} ]},
 {% endif %}
     zip_safe=False,
     install_requires=["torch"],

build2cmake/src/templates/xpu/setup.py

@@ -105,17 +105,17 @@ def build_extension(self, ext: CMakeExtension) -> None:
             for filename in os.listdir(extdir / cfg):
                 move(extdir / cfg / filename, extdir / filename)
 
-
+{% set python_name = name | replace('-', '_') %}
 setup(
-    name="{{ name }}",
+    name="{{ python_name }}",
     # The version is just a stub, it's not used by the final build artefact.
     version="0.1.0",
-    ext_modules=[CMakeExtension("{{ name }}.{{ ops_name }}")],
+    ext_modules=[CMakeExtension("{{ python_name }}.{{ ops_name }}")],
     cmdclass={"build_ext": CMakeBuild},
-    packages=find_packages(where="torch-ext", include=["{{ name }}*"]),
+    packages=find_packages(where="torch-ext", include=["{{ python_name }}*"]),
     package_dir={"": "torch-ext"},
 {% if data_globs %}
-    package_data={"{{ name }}": [ {{ data_globs }} ]},
+    package_data={"{{ python_name }}": [ {{ data_globs }} ]},
 {% endif %}
     zip_safe=False,
     install_requires=["torch"],