bugfix.

2025-11-13 22:17:36 +08:00
parent 3a297646cd
commit f259a0eaf0
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -93,7 +93,7 @@ if __name__ == '__main__':
            ]
        },
        ext_modules=[
-            CUDAExtension(name='deep_gemm_cpp',
+            CUDAExtension(name='deep_gemm.deep_gemm_cpp',
                          sources=sources,
                          include_dirs=build_include_dirs,
                          libraries=build_libraries,