move PATH overwrite to prepare_fbgemm_gpu_build

ROCm · Dec 12, 2023 · fa668e8 · fa668e8
1 parent fb6d601
commit fa668e8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/.github/scripts/fbgemm_gpu_build.bash b/.github/scripts/fbgemm_gpu_build.bash
@@ -43,6 +43,18 @@ prepare_fbgemm_gpu_build () {
   # shellcheck disable=SC2155
   local env_prefix=$(env_name_or_prefix "${env_name}")
 
+  echo "==============old path"
+  echo $PATH
+  new_path="$PATH:/opt/rocm/llvm/bin:\
+/opt/rocm/opencl/bin:\
+/opt/rocm/hip/bin:\
+/opt/rocm/hcc/bin:\
+/opt/rocm/bin
+"
+  echo "==============new path"
+  echo $new_path
+  print_exec conda env config vars set ${env_prefix} PATH="$new_path"
+
   echo "[BUILD] Installing other build dependencies ..."
   # shellcheck disable=SC2086
   (exec_with_retries 3 conda run --no-capture-output ${env_prefix} python -m pip install -r requirements.txt) || return 1

diff --git a/.github/scripts/utils_pytorch.bash b/.github/scripts/utils_pytorch.bash
@@ -128,18 +128,6 @@ install_pytorch_pip () {
   # shellcheck disable=SC2155
   local env_prefix=$(env_name_or_prefix "${env_name}")
 
-  echo "==============old path"
-  echo $PATH
-  new_path="$PATH:/opt/rocm/llvm/bin:\
-/opt/rocm/opencl/bin:\
-/opt/rocm/hip/bin:\
-/opt/rocm/hcc/bin:\
-/opt/rocm/bin
-"
-  echo "==============new path"
-  echo $new_path
-  print_exec conda env config vars set ${env_prefix} PATH="$new_path"
-
   # Install the package from PyTorch PIP (not PyPI)
   install_from_pytorch_pip "${env_name}" torch "${pytorch_version}" "${pytorch_variant_type}" "${pytorch_variant_version}" || return 1