Reset CPU vector lengths

SeisSol · Sep 19, 2023 · a4f0f2b · a4f0f2b
1 parent 8feec4e
commit a4f0f2b
Showing 1 changed file with 17 additions and 16 deletions.
diff --git a/yateto/arch.py b/yateto/arch.py
@@ -115,29 +115,30 @@ def _get_name_and_precision(ident):
 def getArchitectureIdentifiedBy(ident):
   name, precision = _get_name_and_precision(ident)
 
-  # NOTE: ibxsmm currently supports prefetch only for KNL kernels
+  # NOTE: libxsmm currently supports prefetch only for KNL kernels
   arch = {
-    'noarch': Architecture(name, precision, 64, False),
-    'wsm': Architecture(name, precision, 64, False),
-    'snb': Architecture(name, precision, 64, False),
-    'hsw': Architecture(name, precision, 64, False),
+    'noarch': Architecture(name, precision, 16, False),
+    'wsm': Architecture(name, precision, 16, False),
+    'snb': Architecture(name, precision, 32, False),
+    'hsw': Architecture(name, precision, 32, False),
     'skx': Architecture(name, precision, 64, True),
     'knc': Architecture(name, precision, 64, False),
     'knl': Architecture(name, precision, 64, True),
-    'naples': Architecture(name, precision, 64, False),
-    'rome': Architecture(name, precision, 64, False),
-    'milan': Architecture(name, precision, 64, False),
-    'bergamo': Architecture(name, precision, 64, False),
-    'thunderx2t99': Architecture(name, precision, 64, False),
-    'a64fx': Architecture(name, precision, 256, True),
-    'neon': Architecture(name, precision, 64, False),
-    'apple-m1': Architecture(name, precision, 128, False),
-    'sve128': Architecture(name, precision, 64, False),
-    'sve256': Architecture(name, precision, 64, False),
+    'naples': Architecture(name, precision, 32, False),
+    'rome': Architecture(name, precision, 32, False),
+    'milan': Architecture(name, precision, 32, False),
+    'bergamo': Architecture(name, precision, 64, True),
+    'thunderx2t99': Architecture(name, precision, 16, False),
+    'a64fx': Architecture(name, precision, 64, True),
+    'neon': Architecture(name, precision, 16, False),
+    'apple-m1': Architecture(name, precision, 16, False),
+    'sve128': Architecture(name, precision, 16, False),
+    'sve256': Architecture(name, precision, 32, False),
     'sve512': Architecture(name, precision, 64, False),
     'sve1024': Architecture(name, precision, 128, False),
     'sve2048': Architecture(name, precision, 256, False),
-    'power9': Architecture(name, precision, 64, False)
+    'apple-m1': Architecture(name, precision, 16, False),
+    'power9': Architecture(name, precision, 16, False),
   }
   return arch[name]