From 12ac5b8025eb0532197c66312eeb00549a84539c Mon Sep 17 00:00:00 2001
From: Joseph Macaranas <145489236+amd-jmacaran@users.noreply.github.com>
Date: Fri, 21 Mar 2025 17:22:29 -0400
Subject: [PATCH] [External CI] setuptools version fix for pytorch (#4522)

---
 .azuredevops/nightly/pytorch.yml                    | 8 +++++---
 .azuredevops/templates/steps/dependencies-other.yml | 2 +-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/.azuredevops/nightly/pytorch.yml b/.azuredevops/nightly/pytorch.yml
index 56099e78a..6830a9a60 100644
--- a/.azuredevops/nightly/pytorch.yml
+++ b/.azuredevops/nightly/pytorch.yml
@@ -74,7 +74,7 @@ parameters:
     - psutil
     - pyyaml
     - requests
-    - setuptools
+    - setuptools==75.8.0
     - types-dataclasses
     - typing-extensions>=4.8.0
     - sympy>=1.13.0
@@ -170,11 +170,11 @@ jobs:
   - template: /.azuredevops/variables-global.yml
 # various flags/parameters expected by bash scripts in pytorch repo's .ci directory
   - name: ROCM_VERSION
-    value: 6.3.0
+    value: 6.4.0
   - name: ROCM_PATH
     value: /opt/rocm
   - name: DESIRED_CUDA
-    value: 6.3.0
+    value: 6.4.0
   - name: MKLROOT
     value: /opt/intel
   - name: DESIRED_PYTHON
@@ -218,6 +218,7 @@ jobs:
       dependencyList: ${{ parameters.rocmDependencies }}
       dependencySource: staging
       gpuTarget: $(JOB_GPU_TARGET)
+      setupHIPLibrarySymlinks: true
   - task: Bash@3
     displayName: ROCm symbolic link
     inputs:
@@ -279,6 +280,7 @@ jobs:
       targetType: inline
       script: >-
         sudo
+        PYTHONPATH=/home/AzDevOps/.local/lib/python3.10/site-packages:/usr/lib/python3/dist-packages
         DESIRED_CUDA=$(DESIRED_CUDA)
         PYTORCH_ROCM_ARCH=$(JOB_GPU_TARGET)
         GPU_TARGET=$(JOB_GPU_TARGET)
diff --git a/.azuredevops/templates/steps/dependencies-other.yml b/.azuredevops/templates/steps/dependencies-other.yml
index 9f9eca2dc..03ddeee57 100644
--- a/.azuredevops/templates/steps/dependencies-other.yml
+++ b/.azuredevops/templates/steps/dependencies-other.yml
@@ -61,4 +61,4 @@ steps:
     displayName: 'pip install  ...'
     inputs:
       targetType: inline
-      script: pip install -v ${{ join(' ', parameters.pipModules) }}
+      script: pip install -v --force-reinstall ${{ join(' ', parameters.pipModules) }}