fix a small bug

FindDefinition · Jun 15, 2023 · b816f6f · b816f6f
1 parent 7d7b9b2
commit b816f6f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,4 +1,7 @@
 # Changelog
+## [0.4.10] - 2023-06-15
+### Fixed
+- fix a bug in when compile code with arch < sm_75
 
 ## [0.4.9] - 2023-04-06
 ### Added

diff --git a/cumm/gemm/arch/memory.py b/cumm/gemm/arch/memory.py
@@ -99,7 +99,7 @@ def run(self):
             }}
             """)
         else:
-            with code.macro_if_("CUDA_VERSION >= 11040"):
+            with code.macro_if_("CUDA_VERSION >= 11040 && (__CUDA_ARCH__ >= 750)"):
                 self._run(code, self.level, self.prefetch_size)
             with code.macro_else_():
                 self._run(code)

diff --git a/version.txt b/version.txt
@@ -1 +1 @@
-0.4.9
+0.4.10