Set proper AMD architectures.

LBANN · Aug 16, 2024 · 16777c5 · 16777c5
1 parent 3366b27
commit 16777c5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.gitlab/setup_env.sh b/.gitlab/setup_env.sh
@@ -83,8 +83,7 @@ case "${cluster}" in
             extra_rpaths="${ROCM_PATH}/lib:${ROCM_PATH}/llvm/lib:${extra_rpaths}"
         fi
         rocm_platform=ON
-        gpu_arch=gfx90a
-#        gpu_arch=gfx90a,gfx942
+        gpu_arch=gfx90a,gfx942
         launcher=flux
         ROCM_VER=$(basename ${ROCM_PATH})
         PE_ENV_lc=$(echo "${PE_ENV}" | tr '[:upper:]' '[:lower:]')