Skip to content

Commit 6aed36e

Browse files
gs-olivegs-olive
committed
test: Regular CI
1 parent 9b13f10 commit 6aed36e

File tree

6 files changed

+22
-28
lines changed

6 files changed

+22
-28
lines changed

‎.github/workflows/build-test.yml

+12-12
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ jobs:
2020
package-type: wheel
2121
os: linux
2222
test-infra-repository: pytorch/test-infra
23-
test-infra-ref: main
23+
test-infra-ref: 069667b
2424
with-rocm: false
2525
with-cpu: false
2626

@@ -37,12 +37,12 @@ jobs:
3737
smoke-test-script: ""
3838
package-name: torch_tensorrt
3939
name: Build torch-tensorrt whl package
40-
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
40+
uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@069667b
4141
with:
4242
repository: ${{ matrix.repository }}
4343
ref: ""
4444
test-infra-repository: pytorch/test-infra
45-
test-infra-ref: main
45+
test-infra-ref: 069667b
4646
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
4747
pre-script: ${{ matrix.pre-script }}
4848
env-var-script: ${{ matrix.env-var-script }}
@@ -70,7 +70,7 @@ jobs:
7070
repository: "pytorch/tensorrt"
7171
ref: ""
7272
test-infra-repository: pytorch/test-infra
73-
test-infra-ref: main
73+
test-infra-ref: 069667b
7474
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
7575
pre-script: ${{ matrix.pre-script }}
7676
script: |
@@ -99,13 +99,13 @@ jobs:
9999
- repository: pytorch/tensorrt
100100
package-name: torch_tensorrt
101101
pre-script: packaging/pre_build_script.sh
102-
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
102+
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@069667b
103103
with:
104104
job-name: tests-py-dynamo-converters
105105
repository: "pytorch/tensorrt"
106106
ref: ""
107107
test-infra-repository: pytorch/test-infra
108-
test-infra-ref: main
108+
test-infra-ref: 069667b
109109
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
110110
pre-script: ${{ matrix.pre-script }}
111111
script: |
@@ -126,13 +126,13 @@ jobs:
126126
- repository: pytorch/tensorrt
127127
package-name: torch_tensorrt
128128
pre-script: packaging/pre_build_script.sh
129-
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
129+
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@069667b
130130
with:
131131
job-name: tests-py-dynamo-fe
132132
repository: "pytorch/tensorrt"
133133
ref: ""
134134
test-infra-repository: pytorch/test-infra
135-
test-infra-ref: main
135+
test-infra-ref: 069667b
136136
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
137137
pre-script: ${{ matrix.pre-script }}
138138
script: |
@@ -155,13 +155,13 @@ jobs:
155155
- repository: pytorch/tensorrt
156156
package-name: torch_tensorrt
157157
pre-script: packaging/pre_build_script.sh
158-
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
158+
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@069667b
159159
with:
160160
job-name: tests-py-torch-compile-be
161161
repository: "pytorch/tensorrt"
162162
ref: ""
163163
test-infra-repository: pytorch/test-infra
164-
test-infra-ref: main
164+
test-infra-ref: 069667b
165165
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
166166
pre-script: ${{ matrix.pre-script }}
167167
script: |
@@ -183,13 +183,13 @@ jobs:
183183
- repository: pytorch/tensorrt
184184
package-name: torch_tensorrt
185185
pre-script: packaging/pre_build_script.sh
186-
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@main
186+
uses: pytorch/tensorrt/.github/workflows/linux-test.yml@069667b
187187
with:
188188
job-name: tests-py-dynamo-core
189189
repository: "pytorch/tensorrt"
190190
ref: ""
191191
test-infra-repository: pytorch/test-infra
192-
test-infra-ref: main
192+
test-infra-ref: 069667b
193193
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
194194
pre-script: ${{ matrix.pre-script }}
195195
script: |

‎py/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# torch_tensorrt
22

3-
> Ahead of Time (AOT) compiling for PyTorch JIT
3+
> Ahead of Time (AOT) compiling for PyTorch JIT
44
55
Torch-TensorRT is a compiler for PyTorch/TorchScript, targeting NVIDIA GPUs via NVIDIA's TensorRT Deep Learning Optimizer and Runtime. Unlike PyTorch's Just-In-Time (JIT) compiler, Torch-TensorRT is an Ahead-of-Time (AOT) compiler, meaning that before you deploy your TorchScript code, you go through an explicit compile step to convert a standard TorchScript program into an module targeting a TensorRT engine. Torch-TensorRT operates as a PyTorch extention and compiles modules that integrate into the JIT runtime seamlessly. After compilation using the optimized graph should feel no different than running a TorchScript module. You also have access to TensorRT's suite of configurations at compile time, so you are able to specify operating precision (FP32/FP16/INT8) and other settings for your module.
66

‎pyproject.toml

+2-2
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ requires = [
99
"typing-extensions>=4.7.0",
1010
"future>=0.18.3",
1111
"tensorrt>=8.6,<8.7",
12-
"torch>=2.1.0,<2.2.0",
12+
"torch==2.1.1",
1313
"pybind11==2.6.2",
1414
"numpy",
1515
]
@@ -41,7 +41,7 @@ readme = {file = "py/README.md", content-type = "text/markdown"}
4141
requires-python = ">=3.8"
4242
keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
4343
dependencies = [
44-
"torch>=2.1.0,<2.2.0",
44+
"torch==2.1.1",
4545
"tensorrt>=8.6,<8.7",
4646
"packaging>=23",
4747
"numpy",

‎toolchains/ci_workspaces/WORKSPACE.x86_64.cu118.release.rhel

+2-4
Original file line numberDiff line numberDiff line change
@@ -58,17 +58,15 @@ new_local_repository(
5858
http_archive(
5959
name = "libtorch",
6060
build_file = "@//third_party/libtorch:BUILD",
61-
sha256 = "174579a7ee2a506d063714160c5fc57da428f7935311ef511c8f19820eb14c86",
6261
strip_prefix = "libtorch",
63-
urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.0.dev20230731%2Bcu121.zip"],
62+
urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.1%2Bcu121.zip"],
6463
)
6564

6665
http_archive(
6766
name = "libtorch_pre_cxx11_abi",
6867
build_file = "@//third_party/libtorch:BUILD",
69-
sha256 = "532217063c65354d5534211badadc9c370d889cb1c3fdb295c9b3d0f181bc0ba",
7068
strip_prefix = "libtorch",
71-
urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-shared-with-deps-2.1.0.dev20230731%2Bcu121.zip"],
69+
urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.1.1%2Bcu121.zip"],
7270
)
7371

7472
####################################################################################

‎toolchains/ci_workspaces/WORKSPACE.x86_64.cu121.release.rhel

+2-4
Original file line numberDiff line numberDiff line change
@@ -58,17 +58,15 @@ new_local_repository(
5858
http_archive(
5959
name = "libtorch",
6060
build_file = "@//third_party/libtorch:BUILD",
61-
sha256 = "174579a7ee2a506d063714160c5fc57da428f7935311ef511c8f19820eb14c86",
6261
strip_prefix = "libtorch",
63-
urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.0.dev20230731%2Bcu121.zip"],
62+
urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.1%2Bcu121.zip"],
6463
)
6564

6665
http_archive(
6766
name = "libtorch_pre_cxx11_abi",
6867
build_file = "@//third_party/libtorch:BUILD",
69-
sha256 = "532217063c65354d5534211badadc9c370d889cb1c3fdb295c9b3d0f181bc0ba",
7068
strip_prefix = "libtorch",
71-
urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-shared-with-deps-2.1.0.dev20230731%2Bcu121.zip"],
69+
urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.1.1%2Bcu121.zip"],
7270
)
7371

7472
####################################################################################

‎toolchains/ci_workspaces/WORKSPACE.x86_64.release.ubuntu

+3-5
Original file line numberDiff line numberDiff line change
@@ -58,17 +58,15 @@ new_local_repository(
5858
http_archive(
5959
name = "libtorch",
6060
build_file = "@//third_party/libtorch:BUILD",
61-
sha256 = "174579a7ee2a506d063714160c5fc57da428f7935311ef511c8f19820eb14c86",
6261
strip_prefix = "libtorch",
63-
urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.0.dev20230731%2Bcu121.zip"],
62+
urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.1%2Bcu121.zip"],
6463
)
6564

6665
http_archive(
6766
name = "libtorch_pre_cxx11_abi",
6867
build_file = "@//third_party/libtorch:BUILD",
69-
sha256 = "532217063c65354d5534211badadc9c370d889cb1c3fdb295c9b3d0f181bc0ba",
7068
strip_prefix = "libtorch",
71-
urls = ["https://download.pytorch.org/libtorch/nightly/cu121/libtorch-shared-with-deps-2.1.0.dev20230731%2Bcu121.zip"],
69+
urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.1.1%2Bcu121.zip"],
7270
)
7371

7472
####################################################################################
@@ -100,4 +98,4 @@ pip_parse(
10098

10199
load("@devtools_deps//:requirements.bzl", "install_deps")
102100

103-
install_deps()
101+
install_deps()

0 commit comments

Comments
 (0)