Skip to content

Commit a0b1989

Browse files
committed
add test tensorrt workflow
1 parent 6d40ff1 commit a0b1989

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

.github/workflows/build-test-linux.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name: Build and test Linux wheels
22

33
on:
4-
pull_request:
4+
# pull_request:
55
push:
66
branches:
77
- main

.github/workflows/build-test-windows.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name: Build and test Windows wheels
22

33
on:
4-
pull_request:
4+
# pull_request:
55
push:
66
branches:
77
- main

toolchains/ci_workspaces/MODULE.bazel.tmpl

+6-6
Original file line numberDiff line numberDiff line change
@@ -67,20 +67,20 @@ http_archive(
6767
http_archive(
6868
name = "tensorrt",
6969
build_file = "@//third_party/tensorrt/archive:BUILD",
70-
sha256 = "adff1cd5abe5d87013806172351e58fd024e5bf0fc61d49ef4b84cd38ed99081",
71-
strip_prefix = "TensorRT-10.3.0.26",
70+
sha256 = "${TENSORRT_SHA256}",
71+
strip_prefix = "${TENSORRT_STRIP_PREFIX}",
7272
urls = [
73-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.3.0/tars/TensorRT-10.3.0.26.Linux.x86_64-gnu.cuda-12.5.tar.gz",
73+
${TENSORRT_URLS},
7474
],
7575
)
7676

7777
http_archive(
7878
name = "tensorrt_win",
7979
build_file = "@//third_party/tensorrt/archive:BUILD",
80-
sha256 = "2bb4bcb79e8c33575816d874b0512ea28c302af1c06ee6d224da71aa182f75e0",
81-
strip_prefix = "TensorRT-10.3.0.26",
80+
sha256 = "${TENSORRT_SHA256}",
81+
strip_prefix = "${TENSORRT_STRIP_PREFIX}",
8282
urls = [
83-
"https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.3.0/zip/TensorRT-10.3.0.26.Windows.win10.cuda-12.5.zip",
83+
${TENSORRT_URLS},
8484
],
8585
)
8686

0 commit comments

Comments
 (0)