Template linux test Workspace in CI (#2956)

pytorch · Jun 27, 2024 · d245716 · d245716
1 parent 2a6dda9
commit d245716
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git 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
@@ -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(