From 5d206a6a2b44817ef24ac49467afaba361e2f4df Mon Sep 17 00:00:00 2001 From: Dmitri Smirnov Date: Mon, 25 Mar 2024 16:32:25 -0700 Subject: [PATCH] Fix the arguments to enable internal testing --- .../github/azure-pipelines/linux-gpu-ci-pipeline.yml | 1 + .../github/azure-pipelines/win-gpu-ci-pipeline.yml | 7 +++---- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml index 6273b9cf3ef05..0e885b71b486a 100644 --- a/tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml +++ b/tools/ci_build/github/azure-pipelines/linux-gpu-ci-pipeline.yml @@ -151,6 +151,7 @@ stages: --enable_pybind --build_java \ --use_cache \ --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=75 \ + --cmake_extra_defines onnxruntime_BUILD_UNIT_TESTS=ON \ --cmake_extra_defines onnxruntime_ENABLE_CUDA_EP_INTERNAL_TESTS=ON; \ ccache -sv; \ ccache -z" diff --git a/tools/ci_build/github/azure-pipelines/win-gpu-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/win-gpu-ci-pipeline.yml index 4d505248bc7e3..0e340c1d6c175 100644 --- a/tools/ci_build/github/azure-pipelines/win-gpu-ci-pipeline.yml +++ b/tools/ci_build/github/azure-pipelines/win-gpu-ci-pipeline.yml @@ -43,9 +43,10 @@ stages: EnvSetupScript: setup_env_cuda.bat buildArch: x64 additionalBuildFlags: > - --enable_pybind --build_java --build_nodejs --use_cuda --cuda_home="$(Agent.TempDirectory)\v11.8" - --enable_cuda_profiling + --enable_pybind --build_java --build_nodejs --use_cuda --cuda_home="$(Agent.TempDirectory)\v11.8" + --enable_cuda_profiling --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=86 + --cmake_extra_defines onnxruntime_BUILD_UNIT_TESTS=ON --cmake_extra_defines onnxruntime_ENABLE_CUDA_EP_INTERNAL_TESTS=ON msbuildPlatform: x64 isX86: false @@ -67,7 +68,6 @@ stages: --enable_pybind --enable_training --use_cuda --cuda_home="$(Agent.TempDirectory)\v11.8" --skip_onnx_tests --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=86 - --cmake_extra_defines onnxruntime_ENABLE_CUDA_EP_INTERNAL_TESTS=ON msbuildPlatform: x64 isX86: false job_name_suffix: x64_RelWithDebInfo @@ -108,7 +108,6 @@ stages: --cuda_home="$(Agent.TempDirectory)\v11.8" --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=86 --cmake_extra_defines onnxruntime_BUILD_UNIT_TESTS=OFF - --cmake_extra_defines onnxruntime_ENABLE_CUDA_EP_INTERNAL_TESTS=ON msbuildPlatform: x64 isX86: false job_name_suffix: x64_RelWithDebInfo