Add linux debug training package to nightly pipeline (#13192)

This commit is contained in:
Baiju Meswani 2022-10-01 06:58:43 -07:00 committed by GitHub
parent 68218935b9
commit 0cf17b1921
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 15 deletions

View file

@ -1,16 +1,5 @@
trigger: none
parameters:
- name: DebugLinuxBuild
displayName: Debug Linux Build?
type: boolean
default: false
- name: WindowsBuild
displayName: Windows Build?
type: boolean
default: true
resources:
repositories:
- repository: manylinux
@ -31,7 +20,20 @@ stages:
docker_file: Dockerfile.manylinux2014_training_cuda11_6
agent_pool: Onnxruntime-Linux-GPU
upload_wheel: 'yes'
debug_build: ${{ parameters.DebugLinuxBuild }}
debug_build: false
- template: templates/py-packaging-training-cuda-stage.yml
parameters:
build_py_parameters: --enable_training --update --build
torch_version: '1.11.0'
opset_version: '15'
cuda_version: '11.6'
gcc_version: 11
cmake_cuda_architectures: 37;50;52;60;61;70;75;80;86;87
docker_file: Dockerfile.manylinux2014_training_cuda11_6
agent_pool: Onnxruntime-Linux-GPU
upload_wheel: 'no'
debug_build: true
- template: templates/py-packaging-stage.yml
parameters:
@ -39,7 +41,7 @@ stages:
enable_linux_gpu: false
enable_ubuntu_cpu: false
enable_linux_cpu: false
enable_windows_cpu: ${{ parameters.WindowsBuild }}
enable_windows_cpu: true
enable_windows_gpu: false
enable_mac_cpu: false
enable_mac_silicon: false

View file

@ -53,7 +53,7 @@ parameters:
default: false
stages:
- stage: Cuda_Python_Packaging
- stage: "Cuda_Python_Packaging_debug_${{ parameters.debug_build }}"
variables:
- name: isMain
@ -69,6 +69,8 @@ stages:
${{ else }}:
value: 'Release'
dependsOn: []
jobs:
- job: Linux_py_Training_Cuda_Wheels
timeoutInMinutes: 180
@ -202,7 +204,7 @@ stages:
- task: PublishBuildArtifacts@1
displayName: 'Publish Artifact: ONNXRuntime python wheel and documentation'
inputs:
ArtifactName: onnxruntime_gpu
ArtifactName: "onnxruntime_gpu_${{ variables['buildConfig'] }}"
- task: CmdLine@2
displayName: 'Upload wheel'