Skip to content

Commit e72442f

Browse files
lanluo-nvidialanluo-nvidia
committed
test only: c++ tests
1 parent 6cc2faa commit e72442f

File tree

3 files changed

+80
-47
lines changed

3 files changed

+80
-47
lines changed

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

Lines changed: 1 addition & 1 deletion
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-tensorrt-linux.yml

Lines changed: 78 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,8 @@ jobs:
6767
smoke-test-script: ${{ matrix.smoke-test-script }}
6868
trigger-event: ${{ github.event_name }}
6969

70-
tests-py-torchscript-fe:
71-
name: Test torchscript frontend [Python]
70+
tests-cpp-core-conversion:
71+
name: Test core conversion[CPP]
7272
needs: [generate-tensorrt-matrix, build]
7373
strategy:
7474
fail-fast: false
@@ -81,55 +81,88 @@ jobs:
8181
smoke-test-script: packaging/smoke_test_script.sh
8282
uses: ./.github/workflows/linux-test.yml
8383
with:
84-
job-name: tests-py-torchscript-fe
84+
job-name: tests-cpp-core-conversion
8585
repository: "pytorch/tensorrt"
8686
ref: ""
8787
test-infra-repository: pytorch/test-infra
8888
test-infra-ref: main
8989
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
9090
pre-script: ${{ matrix.pre-script }}
9191
script: |
92+
set -x
9293
export USE_HOST_DEPS=1
93-
export CI_BUILD=1
94-
export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
95-
pushd .
96-
cd tests/modules
97-
python hub.py
98-
popd
99-
pushd .
100-
cd tests/py/ts
101-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
102-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
103-
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
104-
popd
94+
python -m pip show torch_tensorrt
95+
location=$(python -m pip show torch_tensorrt | grep Location | sed "s/Location: //g")
96+
ls -lart $location | grep torch_tensorrt
97+
echo $LD_LIBRARY_PATH
98+
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${location}/torch/lib:/usr/local/cuda/lib64:/usr/lib64
99+
echo $LD_LIBRARY_PATH
100+
bazel test //tests/core/conversion:conversion_tests --compilation_mode=opt \
101+
--test_output=summary --config use_precompiled_torchtrt --config pre_cxx11_abi --sandbox_debug --verbose_failures --cxxopt='-std=c++17'
105102
106-
tests-py-dynamo-core:
107-
name: Test dynamo core [Python]
108-
needs: [generate-tensorrt-matrix, build]
109-
strategy:
110-
fail-fast: false
111-
matrix:
112-
include:
113-
- repository: pytorch/tensorrt
114-
package-name: torch_tensorrt
115-
pre-script: packaging/pre_build_script.sh
116-
post-script: packaging/post_build_script.sh
117-
smoke-test-script: packaging/smoke_test_script.sh
118-
uses: ./.github/workflows/linux-test.yml
119-
with:
120-
job-name: tests-py-dynamo-core
121-
repository: "pytorch/tensorrt"
122-
ref: ""
123-
test-infra-repository: pytorch/test-infra
124-
test-infra-ref: main
125-
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
126-
pre-script: ${{ matrix.pre-script }}
127-
script: |
128-
export USE_HOST_DEPS=1
129-
export CI_BUILD=1
130-
pushd .
131-
cd tests/py/dynamo
132-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml runtime/test_000_*
133-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
134-
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
135-
popd
103+
# tests-py-torchscript-fe:
104+
# name: Test torchscript frontend [Python]
105+
# needs: [generate-tensorrt-matrix, build]
106+
# strategy:
107+
# fail-fast: false
108+
# matrix:
109+
# include:
110+
# - repository: pytorch/tensorrt
111+
# package-name: torch_tensorrt
112+
# pre-script: packaging/pre_build_script.sh
113+
# post-script: packaging/post_build_script.sh
114+
# smoke-test-script: packaging/smoke_test_script.sh
115+
# uses: ./.github/workflows/linux-test.yml
116+
# with:
117+
# job-name: tests-py-torchscript-fe
118+
# repository: "pytorch/tensorrt"
119+
# ref: ""
120+
# test-infra-repository: pytorch/test-infra
121+
# test-infra-ref: main
122+
# build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
123+
# pre-script: ${{ matrix.pre-script }}
124+
# script: |
125+
# export USE_HOST_DEPS=1
126+
# export CI_BUILD=1
127+
# export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
128+
# pushd .
129+
# cd tests/modules
130+
# python hub.py
131+
# popd
132+
# pushd .
133+
# cd tests/py/ts
134+
# python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
135+
# python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
136+
# python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
137+
# popd
138+
139+
# tests-py-dynamo-core:
140+
# name: Test dynamo core [Python]
141+
# needs: [generate-tensorrt-matrix, build]
142+
# strategy:
143+
# fail-fast: false
144+
# matrix:
145+
# include:
146+
# - repository: pytorch/tensorrt
147+
# package-name: torch_tensorrt
148+
# pre-script: packaging/pre_build_script.sh
149+
# post-script: packaging/post_build_script.sh
150+
# smoke-test-script: packaging/smoke_test_script.sh
151+
# uses: ./.github/workflows/linux-test.yml
152+
# with:
153+
# job-name: tests-py-dynamo-core
154+
# repository: "pytorch/tensorrt"
155+
# ref: ""
156+
# test-infra-repository: pytorch/test-infra
157+
# test-infra-ref: main
158+
# build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
159+
# pre-script: ${{ matrix.pre-script }}
160+
# script: |
161+
# export USE_HOST_DEPS=1
162+
# export CI_BUILD=1
163+
# pushd .
164+
# cd tests/py/dynamo
165+
# python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml runtime/test_000_*
166+
# python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
167+
# python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
168+
# popd

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

Lines changed: 1 addition & 1 deletion
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

0 commit comments

Comments
 (0)