From 1cb743bd2313ca5d1845dbdda6cccd9dcaa5d30c Mon Sep 17 00:00:00 2001 From: Dick Carter Date: Sun, 3 Mar 2019 15:23:38 -0800 Subject: [PATCH] Add test temporarily to tests run with tensorrt CI build (cuda10, cudnn7.4.2) --- ci/docker/runtime_functions.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh index de1b7795ce69..848f15267270 100755 --- a/ci/docker/runtime_functions.sh +++ b/ci/docker/runtime_functions.sh @@ -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