Skip to content

test only: c++ tests #3283

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 2 commits into from
Closed

test only: c++ tests #3283

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e72442f
test only: c++ tests
lanluo-nvidia Nov 6, 2024
ce45ffa
test
lanluo-nvidia Nov 6, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/build-test-linux.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name: Build and test Linux wheels

on:
pull_request:
# pull_request:
push:
branches:
- main
Expand Down
124 changes: 79 additions & 45 deletions .github/workflows/build-test-tensorrt-linux.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
name: Build and Test Torch-TensorRT on Linux with Future TensorRT Versions

on:
pull_request:
workflow_dispatch:

permissions:
Expand Down Expand Up @@ -67,8 +68,8 @@ jobs:
smoke-test-script: ${{ matrix.smoke-test-script }}
trigger-event: ${{ github.event_name }}

tests-py-torchscript-fe:
name: Test torchscript frontend [Python]
tests-cpp-core-conversion:
name: Test core conversion[CPP]
needs: [generate-tensorrt-matrix, build]
strategy:
fail-fast: false
Expand All @@ -81,55 +82,88 @@ jobs:
smoke-test-script: packaging/smoke_test_script.sh
uses: ./.github/workflows/linux-test.yml
with:
job-name: tests-py-torchscript-fe
job-name: tests-cpp-core-conversion
repository: "pytorch/tensorrt"
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
pre-script: ${{ matrix.pre-script }}
script: |
set -x
export USE_HOST_DEPS=1
export CI_BUILD=1
export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
pushd .
cd tests/modules
python hub.py
popd
pushd .
cd tests/py/ts
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
popd
python -m pip show torch_tensorrt
location=$(python -m pip show torch_tensorrt | grep Location | sed "s/Location: //g")
ls -lart $location | grep torch_tensorrt
echo $LD_LIBRARY_PATH
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${location}/torch/lib:/usr/local/cuda/lib64:/usr/lib64
echo $LD_LIBRARY_PATH
bazel test //tests/core/conversion:conversion_tests --compilation_mode=opt \
--test_output=summary --config use_precompiled_torchtrt --config pre_cxx11_abi --sandbox_debug --verbose_failures --cxxopt='-std=c++17'

tests-py-dynamo-core:
name: Test dynamo core [Python]
needs: [generate-tensorrt-matrix, build]
strategy:
fail-fast: false
matrix:
include:
- repository: pytorch/tensorrt
package-name: torch_tensorrt
pre-script: packaging/pre_build_script.sh
post-script: packaging/post_build_script.sh
smoke-test-script: packaging/smoke_test_script.sh
uses: ./.github/workflows/linux-test.yml
with:
job-name: tests-py-dynamo-core
repository: "pytorch/tensorrt"
ref: ""
test-infra-repository: pytorch/test-infra
test-infra-ref: main
build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
pre-script: ${{ matrix.pre-script }}
script: |
export USE_HOST_DEPS=1
export CI_BUILD=1
pushd .
cd tests/py/dynamo
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml runtime/test_000_*
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
popd
# tests-py-torchscript-fe:
# name: Test torchscript frontend [Python]
# needs: [generate-tensorrt-matrix, build]
# strategy:
# fail-fast: false
# matrix:
# include:
# - repository: pytorch/tensorrt
# package-name: torch_tensorrt
# pre-script: packaging/pre_build_script.sh
# post-script: packaging/post_build_script.sh
# smoke-test-script: packaging/smoke_test_script.sh
# uses: ./.github/workflows/linux-test.yml
# with:
# job-name: tests-py-torchscript-fe
# repository: "pytorch/tensorrt"
# ref: ""
# test-infra-repository: pytorch/test-infra
# test-infra-ref: main
# build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
# pre-script: ${{ matrix.pre-script }}
# script: |
# export USE_HOST_DEPS=1
# export CI_BUILD=1
# export LD_LIBRARY_PATH=/usr/lib64:$LD_LIBRARY_PATH
# pushd .
# cd tests/modules
# python hub.py
# popd
# pushd .
# cd tests/py/ts
# python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_api_test_results.xml api/
# python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_models_test_results.xml models/
# python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/ts_integrations_test_results.xml integrations/
# popd

# tests-py-dynamo-core:
# name: Test dynamo core [Python]
# needs: [generate-tensorrt-matrix, build]
# strategy:
# fail-fast: false
# matrix:
# include:
# - repository: pytorch/tensorrt
# package-name: torch_tensorrt
# pre-script: packaging/pre_build_script.sh
# post-script: packaging/post_build_script.sh
# smoke-test-script: packaging/smoke_test_script.sh
# uses: ./.github/workflows/linux-test.yml
# with:
# job-name: tests-py-dynamo-core
# repository: "pytorch/tensorrt"
# ref: ""
# test-infra-repository: pytorch/test-infra
# test-infra-ref: main
# build-matrix: ${{ needs.generate-tensorrt-matrix.outputs.matrix }}
# pre-script: ${{ matrix.pre-script }}
# script: |
# export USE_HOST_DEPS=1
# export CI_BUILD=1
# pushd .
# cd tests/py/dynamo
# python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_runtime_test_results.xml runtime/test_000_*
# python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_partitioning_test_results.xml partitioning/
# python -m pytest -ra -n 4 --junitxml=${RUNNER_TEST_RESULTS_DIR}/tests_py_dynamo_core_lowering_test_results.xml lowering/
# popd
2 changes: 1 addition & 1 deletion .github/workflows/build-test-windows.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name: Build and test Windows wheels

on:
pull_request:
# pull_request:
push:
branches:
- main
Expand Down
Loading