From 8ae59885b416a75d307d30c0c9079f1450b386de Mon Sep 17 00:00:00 2001 From: "Wang, Yi A" Date: Tue, 25 Nov 2025 12:58:48 +0000 Subject: [PATCH] fix pip3 install error for example like cutlass-gemm in local build. Signed-off-by: Wang, Yi A --- build2cmake/src/templates/cpu/setup.py | 9 +++++---- build2cmake/src/templates/cuda/setup.py | 9 +++++---- build2cmake/src/templates/metal/setup.py | 9 +++++---- build2cmake/src/templates/xpu/setup.py | 10 +++++----- 4 files changed, 20 insertions(+), 17 deletions(-) diff --git a/build2cmake/src/templates/cpu/setup.py b/build2cmake/src/templates/cpu/setup.py index 08164f1a..60d4853c 100644 --- a/build2cmake/src/templates/cpu/setup.py +++ b/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"], diff --git a/build2cmake/src/templates/cuda/setup.py b/build2cmake/src/templates/cuda/setup.py index 0d3c541c..69808b37 100644 --- a/build2cmake/src/templates/cuda/setup.py +++ b/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"], diff --git a/build2cmake/src/templates/metal/setup.py b/build2cmake/src/templates/metal/setup.py index 08164f1a..60d4853c 100644 --- a/build2cmake/src/templates/metal/setup.py +++ b/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"], diff --git a/build2cmake/src/templates/xpu/setup.py b/build2cmake/src/templates/xpu/setup.py index 55bedcaf..6289c99b 100644 --- a/build2cmake/src/templates/xpu/setup.py +++ b/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"],