Fix when local 6.2.0 MIOpen library is linked in.

LBANN · Aug 20, 2024 · 84d1de7 · 84d1de7
1 parent 70b0645
commit 84d1de7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.gitlab/setup_env.sh b/.gitlab/setup_env.sh
@@ -118,7 +118,10 @@ fi
 #CMAKE_PREFIX_PATH=${INSTALL_EXTERNALS_ROOT}/${SYSTEM_INSTALL_PREFIX_EXTERNALS}
 case "${cluster}" in
     tioga)
-        CMAKE_PREFIX_PATH=/p/vast1/lbann/stable_dependencies/tioga/rocm-6.2.0/miopen:${CMAKE_PREFIX_PATH}
+        ROCM_VER=$(basename ${ROCM_PATH})
+        if [[ "${ROCM_VER}" = "6.2.0" ]]; then
+            CMAKE_PREFIX_PATH=/p/vast1/lbann/stable_dependencies/tioga/rocm-6.2.0/miopen:${CMAKE_PREFIX_PATH}
+        fi
         ;;
     *)
         ;;