From 241b94031456527448404aeecb59fbf48ee69f37 Mon Sep 17 00:00:00 2001
From: lanluo-nvidia <lanl@nvidia.com>
Date: Sun, 23 Nov 2025 19:53:50 -0800
Subject: [PATCH 1/3] fix ci broken issue

---
 .github/workflows/build-test-linux-aarch64.yml    | 1 +
 .github/workflows/build-test-linux-x86_64.yml     | 1 +
 .github/workflows/build-test-linux-x86_64_rtx.yml | 1 +
 3 files changed, 3 insertions(+)

diff --git a/.github/workflows/build-test-linux-aarch64.yml b/.github/workflows/build-test-linux-aarch64.yml
index 63573c7093..3711337f28 100644
--- a/.github/workflows/build-test-linux-aarch64.yml
+++ b/.github/workflows/build-test-linux-aarch64.yml
@@ -77,6 +77,7 @@ jobs:
       smoke-test-script: ${{ matrix.smoke-test-script }}
       trigger-event: ${{ github.event_name }}
       architecture: "aarch64"
+      pip-install-torch-extra-args: "--extra-index-url https://pypi.org/simple"
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
diff --git a/.github/workflows/build-test-linux-x86_64.yml b/.github/workflows/build-test-linux-x86_64.yml
index da3124c472..bc39981d1f 100644
--- a/.github/workflows/build-test-linux-x86_64.yml
+++ b/.github/workflows/build-test-linux-x86_64.yml
@@ -77,6 +77,7 @@ jobs:
       trigger-event: ${{ github.event_name }}
       architecture: "x86_64"
       use-rtx: false
+      pip-install-torch-extra-args: "--extra-index-url https://pypi.org/simple"
 
   L0-dynamo-converter-tests:
     name: L0 dynamo converter tests
diff --git a/.github/workflows/build-test-linux-x86_64_rtx.yml b/.github/workflows/build-test-linux-x86_64_rtx.yml
index 4c1901ee7c..f627706c91 100644
--- a/.github/workflows/build-test-linux-x86_64_rtx.yml
+++ b/.github/workflows/build-test-linux-x86_64_rtx.yml
@@ -76,6 +76,7 @@ jobs:
       trigger-event: ${{ github.event_name }}
       architecture: "x86_64"
       use-rtx: true
+      pip-install-torch-extra-args: "--extra-index-url https://pypi.org/simple"
 
   L0-dynamo-converter-tests:
     name: L0 dynamo converter tests

From 90248392963952d866a87fcd5c0b9f0831152a50 Mon Sep 17 00:00:00 2001
From: lanluo-nvidia <lanl@nvidia.com>
Date: Sun, 23 Nov 2025 21:16:18 -0800
Subject: [PATCH 2/3] fix ci issue

---
 packaging/pre_build_script.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh
index 4c21fd6fda..bd0227044f 100755
--- a/packaging/pre_build_script.sh
+++ b/packaging/pre_build_script.sh
@@ -53,7 +53,7 @@ else
     INDEX_URL=https://download.pytorch.org/whl/${CHANNEL}/${CU_VERSION}
 
     # Install all the dependencies required for Torch-TensorRT
-    pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL}
+    pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
 fi
 
 export TORCH_BUILD_NUMBER=$(python -c "import torch, urllib.parse as ul; print(ul.quote_plus(torch.__version__))")

From 41146608185284c48c0faa067b13f4333b07179b Mon Sep 17 00:00:00 2001
From: lanluo-nvidia <lanl@nvidia.com>
Date: Sun, 23 Nov 2025 21:57:46 -0800
Subject: [PATCH 3/3] fix ci issue

---
 .github/scripts/install-torch-tensorrt.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/scripts/install-torch-tensorrt.sh b/.github/scripts/install-torch-tensorrt.sh
index dccf4925a2..cf0a6f0940 100755
--- a/.github/scripts/install-torch-tensorrt.sh
+++ b/.github/scripts/install-torch-tensorrt.sh
@@ -18,8 +18,8 @@ pip install --pre -r ${PWD}/tests/py/requirements.txt
 # eg. timm will install the latest torchvision, however we want to use the torchvision from nightly
 # reinstall torch torchvisionto make sure we have the correct version
 pip uninstall -y torch torchvision
-pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL}
-pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL}
+pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
+pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
 
 if [[ ${USE_TRT_RTX} == true ]]; then
     source .github/scripts/install-tensorrt-rtx.sh