diff --git a/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml index 59deb0d4975fe..ccb03ed7a5106 100644 --- a/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml +++ b/tools/ci_build/github/azure-pipelines/bigmodels-ci-pipeline.yml @@ -123,7 +123,7 @@ stages: --enable_cuda_profiling \ --enable_pybind --build_java \ --use_cache \ - --cmake_extra_defines 'CMAKE_CUDA_ARCHITECTURES=75;86' ; \ + --cmake_extra_defines 'CMAKE_CUDA_ARCHITECTURES=75;80' ; \ ccache -sv; \ ccache -z" workingDirectory: $(Build.SourcesDirectory) @@ -413,7 +413,7 @@ stages: skipComponentGovernanceDetection: true workspace: clean: all - pool: Onnxruntime-Linux-A10-24G + pool: Onnxruntime-Linux-GPU-A100-WUS3 steps: - checkout: self 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 518aec8c2f92a..339608e9e04a0 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 @@ -137,7 +137,7 @@ stages: skipComponentGovernanceDetection: true workspace: clean: all - pool: Onnxruntime-Linux-A10-24G + pool: Onnxruntime-Linux-GPU-A100-WUS3 steps: - checkout: self clean: true diff --git a/tools/ci_build/github/linux/build_cuda_ci.sh b/tools/ci_build/github/linux/build_cuda_ci.sh index fbf69d268a033..16c3ab87e5fb7 100755 --- a/tools/ci_build/github/linux/build_cuda_ci.sh +++ b/tools/ci_build/github/linux/build_cuda_ci.sh @@ -21,7 +21,7 @@ BUILD_ARGS=('--config' "--enable_pybind" "--build_java" "--cmake_extra_defines" - "CMAKE_CUDA_ARCHITECTURES=86" + "CMAKE_CUDA_ARCHITECTURES=80" "onnxruntime_BUILD_UNIT_TESTS=ON" "onnxruntime_ENABLE_CUDA_EP_INTERNAL_TESTS=ON") if [ -x "$(command -v ninja)" ]; then