|
35 | 35 | pre-script: packaging/pre_build_script.sh
|
36 | 36 | env-var-script: packaging/env_vars.txt
|
37 | 37 | post-script: ""
|
38 |
| - smoke-test-script: "" |
| 38 | + smoke-test-script: "packaging/smoke_test.py" |
39 | 39 | package-name: torch_tensorrt
|
40 | 40 | name: Build torch-tensorrt whl package
|
41 | 41 | uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.2
|
|
74 | 74 | test-infra-repository: pytorch/test-infra
|
75 | 75 | test-infra-ref: release/2.2
|
76 | 76 | build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
|
| 77 | + smoke-test-script: ${{ matrix.smoke-test-script }} |
77 | 78 | pre-script: ${{ matrix.pre-script }}
|
78 | 79 | script: |
|
79 | 80 | export USE_HOST_DEPS=1
|
@@ -110,6 +111,7 @@ jobs:
|
110 | 111 | test-infra-repository: pytorch/test-infra
|
111 | 112 | test-infra-ref: release/2.2
|
112 | 113 | build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
|
| 114 | + smoke-test-script: ${{ matrix.smoke-test-script }} |
113 | 115 | pre-script: ${{ matrix.pre-script }}
|
114 | 116 | script: |
|
115 | 117 | export USE_HOST_DEPS=1
|
@@ -138,6 +140,7 @@ jobs:
|
138 | 140 | test-infra-repository: pytorch/test-infra
|
139 | 141 | test-infra-ref: release/2.2
|
140 | 142 | build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
|
| 143 | + smoke-test-script: ${{ matrix.smoke-test-script }} |
141 | 144 | pre-script: ${{ matrix.pre-script }}
|
142 | 145 | script: |
|
143 | 146 | export USE_HOST_DEPS=1
|
@@ -168,6 +171,7 @@ jobs:
|
168 | 171 | test-infra-repository: pytorch/test-infra
|
169 | 172 | test-infra-ref: release/2.2
|
170 | 173 | build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
|
| 174 | + smoke-test-script: ${{ matrix.smoke-test-script }} |
171 | 175 | pre-script: ${{ matrix.pre-script }}
|
172 | 176 | script: |
|
173 | 177 | export USE_HOST_DEPS=1
|
@@ -197,6 +201,7 @@ jobs:
|
197 | 201 | test-infra-repository: pytorch/test-infra
|
198 | 202 | test-infra-ref: release/2.2
|
199 | 203 | build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
|
| 204 | + smoke-test-script: ${{ matrix.smoke-test-script }} |
200 | 205 | pre-script: ${{ matrix.pre-script }}
|
201 | 206 | script: |
|
202 | 207 | export USE_HOST_DEPS=1
|
|
0 commit comments