Merge pull request #55 from SeisSol/fugaku

Pspamm a64fx
SeisSol · Aug 16, 2023 · c788a85 · c788a85
2 parents a7aef34 + e8575ed
commit c788a85
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/yateto/codegen/gemm/gemmgen.py b/yateto/codegen/gemm/gemmgen.py
@@ -244,7 +244,7 @@ def __call__(self, routineName, fileName):
         self._gemmDescr['alpha'],
         self._gemmDescr['beta'],
         '--arch',
-        cpu_arch,
+        'arm_sve' if cpu_arch == 'a64fx' else cpu_arch, 
         '--prefetching',
         self._gemmDescr['prefetch'],
         '--output_funcname',

diff --git a/yateto/gemm_configuration.py b/yateto/gemm_configuration.py
@@ -210,7 +210,7 @@ def __init__(self, arch, cmd: str = 'pspamm.py', threshold: int = 128):
     self._threshold = threshold
 
   def _archSupported(self):
-    supported_set = {'thunderx2t99', 'knl', 'skx'}
+    supported_set = {'thunderx2t99', 'knl', 'skx', 'a64fx'}
     if self._arch.name.lower() in supported_set:
       return True
     else: