Merge pull request #2343 from devitocodes/sm_cc

arch: Use get_nvidia_cc to get Nvidia GPU architecture
devitocodes · Apr 3, 2024 · 311cd4e · 311cd4e
2 parents 39ea32a + 6e64c16
commit 311cd4e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/devito/arch/compiler.py b/devito/arch/compiler.py
@@ -668,7 +668,11 @@ def __init_finalize__(self, **kwargs):
                     proc_link_flags.append(i)
             self.ldflags.extend(proc_link_flags)
 
-        self.cflags.append('-arch=native')
+        cc = get_nvidia_cc()
+        if cc:
+            self.cflags.append(f'-arch=sm_{cc}')
+        else:
+            self.cflags.append('-arch=native')
 
         # Disable `warning #1650-D: result of call is not used`
         # See `https://gist.github.com/gavinb/f2320f9eaa0e0a7efca6877a34047a9d` about