Fix ROCm issue where target ID is set to gfxgfx*** (#1290)

spcl · Jun 30, 2023 · e66392f · e66392f
1 parent 81b3e4e
commit e66392f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/AUTHORS b/AUTHORS
@@ -35,5 +35,6 @@ Tiancheng Chen
 Reid Wahl
 Yihang Luo
 Alexandru Calotoiu
+Phillip Lane
 
 and other contributors listed in https://github.com/spcl/dace/graphs/contributors
diff --git a/dace/codegen/targets/cuda.py b/dace/codegen/targets/cuda.py
@@ -497,7 +497,7 @@ def cmake_options():
             hip_arch = [ha for ha in hip_arch if ha is not None and len(ha) > 0]
 
             flags = Config.get("compiler", "cuda", "hip_args")
-            flags += ' ' + ' '.join('--offload-arch=gfx{arch}'.format(arch=arch) for arch in hip_arch)
+            flags += ' ' + ' '.join('--offload-arch={arch}'.format(arch=arch if arch.startswith("gfx") else "gfx" + arch) for arch in hip_arch)
             options.append("-DEXTRA_HIP_FLAGS=\"{}\"".format(flags))
 
         if Config.get('compiler', 'cpu', 'executable'):