Skip to content

Commit

Permalink
Add test temporarily to tests run with tensorrt CI build (cuda10, cud…
Browse files Browse the repository at this point in the history
…nn7.4.2)
  • Loading branch information
DickJC123 committed Mar 3, 2019
1 parent 817bad0 commit 1cb743b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -832,6 +832,7 @@ unittest_ubuntu_tensorrt_gpu() {
export CUDNN_VERSION=7.0.3
python tests/python/tensorrt/lenet5_train.py
nosetests-3.4 $NOSE_COVERAGE_ARGUMENTS $NOSE_TIMER_ARGUMENTS --with-xunit --xunit-file nosetests_trt_gpu.xml --verbose --nocapture tests/python/tensorrt/
nosetests-3.4 $NOSE_COVERAGE_ARGUMENTS $NOSE_TIMER_ARGUMENTS --with-xunit --xunit-file nosetests_trt_gpu.xml --verbose --nocapture tests/python/gpu/test_operator_gpu.py:test_conv_deconv_guards
}

# quantization gpu currently only runs on P3 instances
Expand Down

0 comments on commit 1cb743b

Please sign in to comment.