arch: Add more nvidia archs

devitocodes · Jan 24, 2025 · b8de9ec · b8de9ec
1 parent 3d85383
commit b8de9ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/devito/arch/archinfo.py b/devito/arch/archinfo.py
@@ -32,7 +32,7 @@
            # Generic GPUs
            'AMDGPUX', 'NVIDIAX', 'INTELGPUX',
            # Nvidia GPUs
-           'AMPERE',
+           'VOLTA', 'AMPERE', 'HOPPER', 'BLACKWELL',
            # Intel GPUs
            'PVC', 'INTELGPUMAX', 'MAX1100', 'MAX1550']
 
@@ -912,7 +912,7 @@ def supports(self, query, language=None):
             # Asynchronous pipeline loads -- introduced in Ampere
             return True
         elif query == 'tma' and cc >= 90:
-            # Tensor Memory Acceleratory -- introduced in Hopper
+            # Tensor Memory Accelerator -- introduced in Hopper
             return True
         else:
             return False