adapt #19052 (comment)

microsoft · Dec 5, 2024 · fa4dd84 · fa4dd84
1 parent 82877da
commit fa4dd84
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/ci_build/github/linux/build_tensorrt_ci.sh b/tools/ci_build/github/linux/build_tensorrt_ci.sh
@@ -26,8 +26,11 @@ BUILD_ARGS=('--config' 'Release'
 for arg in "$@"; do
   case $arg in
     --cuda_minimal=ON)
+      # Replace onnxruntime_BUILD_UNIT_TESTS=ON with OFF
+      BUILD_ARGS=("${BUILD_ARGS[@]/onnxruntime_BUILD_UNIT_TESTS=ON/onnxruntime_BUILD_UNIT_TESTS=OFF}")
       BUILD_ARGS+=("onnxruntime_CUDA_MINIMAL=ON")
       BUILD_ARGS+=("onnxruntime_DISABLE_CONTRIB_OPS=ON")
+      BUILD_ARGS+=("--skip_tests")
       ;;
   esac
 done