arch: Use get_nvidia_cc to get Nvidia gpu architecture rather than us…

…ing 'native'. This fixes compilation errors on the Nvidia Jetson.
devitocodes · Apr 2, 2024 · 3621e78 · 3621e78
1 parent 8d9cacb
commit 3621e78
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