training CI bug fixes

microsoft · Aug 14, 2024 · 2ccab4f · 2ccab4f
1 parent bb85118
commit 2ccab4f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ci-pipeline.yml b/tools/ci_build/github/azure-pipelines/orttraining-linux-gpu-ci-pipeline.yml
@@ -48,7 +48,7 @@ jobs:
     RunInjectedPipeline: 'true'
     InjectedPipeline: 'orttraining-linux-gpu-test-ci-pipeline.yml'
     DockerImageTag: 'onnxruntime_orttraining_ortmodule_tests_image'
-    TimeoutInMinutes: 140
+    TimeoutInMinutes: 150
     # Enable unreleased onnx opsets in CI builds
     # This facilitates testing the implementation for the new opsets
     AllowReleasedOpsetOnly: '0'
diff --git a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda.yml b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda.yml
@@ -18,7 +18,7 @@ stages:
     torch_version: '2.0.0'
     opset_version: '17'
     cuda_version: '11.8'
-    cmake_cuda_architectures: 60;61;70;75;80;86
+    cmake_cuda_architectures: 70;75;80;86
     docker_file: Dockerfile.manylinux2_28_training_cuda11_8
     agent_pool: Onnxruntime-Linux-GPU
     upload_wheel: 'yes'

diff --git a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda12.yml b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda12.yml
@@ -8,7 +8,7 @@ stages:
     torch_version: '2.1.0'
     opset_version: '17'
     cuda_version: '12.2'
-    cmake_cuda_architectures: 70;75;80;86;90
+    cmake_cuda_architectures: 80;86;90
     docker_file: Dockerfile.manylinux2_28_training_cuda12_2
     agent_pool: Onnxruntime-Linux-GPU
     upload_wheel: 'yes'