saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-08 00:23:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Split the GPU pipeline to 3 different machine pools (#11724)

Browse source
This commit is contained in:
Changming Sun 2022-06-03 10:57:32 -07:00 committed by GitHub
parent 4445dd6bc1
commit ec05313cd9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tools/ci_build/github/azure-pipelines/templates/win-gpu-ci.yml
View file

@ -41,6 +41,9 @@ parameters:
- name: ORT_EP_NAME
type: string
- name: MachinePool
type: string
- name: DocUpdateNeeded
displayName: Run Tests?
@ -58,7 +61,7 @@ jobs:
DocUpdateNeeded: ${{ parameters.DocUpdateNeeded }}
workspace:
clean: all
pool: 'Win-GPU-2019'
pool: ${{ parameters.MachinePool }}
timeoutInMinutes: 300
steps:
- task: UsePythonVersion@0

7
tools/ci_build/github/azure-pipelines/win-gpu-ci-pipeline.yml
View file

@ -20,6 +20,7 @@ stages:
RunOnnxRuntimeTests: ${{ parameters.RunOnnxRuntimeTests }}
RunStaticCodeAnalysis: false
ORT_EP_NAME: CUDA
MachinePool: onnxruntime-Win2019-GPU
DocUpdateNeeded: true
@ -38,6 +39,7 @@ stages:
RunOnnxRuntimeTests: ${{ parameters.RunOnnxRuntimeTests }}
RunStaticCodeAnalysis: false
ORT_EP_NAME: CUDA
MachinePool: onnxruntime-Win2019-GPU-training
isTraining: true
- stage: dml
@ -48,11 +50,12 @@ stages:
BuildConfig: 'RelWithDebInfo'
EnvSetupScript: setup_env.bat
buildArch: x64
additionalBuildFlags: --enable_pybind --use_dml --enable_wcos --use_winml --cmake_extra_defines CMAKE_SYSTEM_VERSION=10.0.18362.0
additionalBuildFlags: --enable_pybind --use_dml --enable_wcos --use_winml
msbuildPlatform: x64
isX86: false
job_name_suffix: x64_RelWithDebInfo
RunOnnxRuntimeTests: ${{ parameters.RunOnnxRuntimeTests }}
RunStaticCodeAnalysis: false
ORT_EP_NAME: DML
ORT_EP_NAME: DML
MachinePool: onnxruntime-Win2019-GPU-dml