From d245716ef96e75ed28a61659aa3625238d209008 Mon Sep 17 00:00:00 2001
From: lanluo-nvidia <lanl@nvidia.com>
Date: Thu, 27 Jun 2024 10:56:35 -0700
Subject: [PATCH] Template linux test Workspace in CI (#2956)

---
 .github/workflows/build-test-linux.yml             | 14 +++++++-------
 .../WORKSPACE.x86_64.release.rhel.tmpl             |  4 ++--
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/build-test-linux.yml b/.github/workflows/build-test-linux.yml
index 22f03e6218..b862673f0b 100644
--- a/.github/workflows/build-test-linux.yml
+++ b/.github/workflows/build-test-linux.yml
@@ -66,7 +66,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
             smoke-test-script: packaging/smoke_test_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: ./.github/workflows/linux-test.yml
     with:
       job-name: tests-py-torchscript-fe
       repository: "pytorch/tensorrt"
@@ -101,7 +101,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
             smoke-test-script: packaging/smoke_test_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: ./.github/workflows/linux-test.yml
     with:
       job-name: tests-py-dynamo-converters
       repository: "pytorch/tensorrt"
@@ -129,7 +129,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
             smoke-test-script: packaging/smoke_test_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: ./.github/workflows/linux-test.yml
     with:
       job-name: tests-py-dynamo-fe
       repository: "pytorch/tensorrt"
@@ -158,7 +158,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
             smoke-test-script: packaging/smoke_test_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: ./.github/workflows/linux-test.yml
     with:
       job-name: tests-py-dynamo-serde
       repository: "pytorch/tensorrt"
@@ -186,7 +186,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
             smoke-test-script: packaging/smoke_test_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: ./.github/workflows/linux-test.yml
     with:
       job-name: tests-py-torch-compile-be
       repository: "pytorch/tensorrt"
@@ -216,7 +216,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
             smoke-test-script: packaging/smoke_test_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
+    uses: ./.github/workflows/linux-test.yml
     with:
       job-name: tests-py-dynamo-core
       repository: "pytorch/tensorrt"
@@ -246,7 +246,7 @@ jobs:
             pre-script: packaging/pre_build_script.sh
             post-script: packaging/post_build_script.sh
             smoke-test-script: packaging/smoke_test_script.sh
-    uses: pytorch/tensorrt/.github/workflows/linux-test.yml@release/2.3
+    uses: ./.github/workflows/linux-test.yml
     with:
       job-name: tests-py-core
       repository: "pytorch/tensorrt"
diff --git a/toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel.tmpl b/toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel.tmpl
index 3dcde3cf56..146ce9e90d 100644
--- a/toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel.tmpl
+++ b/toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel.tmpl
@@ -58,14 +58,14 @@ http_archive(
     name = "libtorch",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/nightly/cu124/libtorch-cxx11-abi-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/${CHANNEL}/${CU_VERSION}/libtorch-cxx11-abi-shared-with-deps-latest.zip"],
 )
 
 http_archive(
     name = "libtorch_pre_cxx11_abi",
     build_file = "@//third_party/libtorch:BUILD",
     strip_prefix = "libtorch",
-    urls = ["https://download.pytorch.org/libtorch/nightly/cu124/libtorch-shared-with-deps-latest.zip"],
+    urls = ["https://download.pytorch.org/libtorch/${CHANNEL}/${CU_VERSION}/libtorch-shared-with-deps-latest.zip"],
 )
 
 http_archive(