mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-02 23:39:58 +00:00
61 lines
2.3 KiB
YAML
61 lines
2.3 KiB
YAML
parameters:
|
|
- name: RunOnnxRuntimeTests
|
|
displayName: Run Tests?
|
|
type: boolean
|
|
default: true
|
|
|
|
stages:
|
|
- stage: cuda
|
|
dependsOn: []
|
|
jobs:
|
|
- template: templates/win-gpu-ci.yml
|
|
parameters:
|
|
BuildConfig: 'RelWithDebInfo'
|
|
EnvSetupScript: setup_env_cuda_11.bat
|
|
buildArch: x64
|
|
additionalBuildFlags: --enable_pybind --build_java --build_nodejs --use_cuda --cuda_version=11.4 --cuda_home="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4" --cudnn_home="C:\local\cudnn-11.4-windows-x64-v8.2.2.26\cuda" --gen_doc validate --enable_cuda_profiling --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=52
|
|
msbuildPlatform: x64
|
|
isX86: false
|
|
job_name_suffix: x64_RelWithDebInfo
|
|
RunOnnxRuntimeTests: ${{ parameters.RunOnnxRuntimeTests }}
|
|
RunStaticCodeAnalysis: false
|
|
ORT_EP_NAME: CUDA
|
|
MachinePool: onnxruntime-Win2019-GPU
|
|
DocUpdateNeeded: true
|
|
|
|
|
|
- stage: training
|
|
dependsOn: []
|
|
jobs:
|
|
- template: templates/win-gpu-ci.yml
|
|
parameters:
|
|
BuildConfig: 'RelWithDebInfo'
|
|
EnvSetupScript: setup_env_cuda_11.bat
|
|
buildArch: x64
|
|
additionalBuildFlags: --enable_pybind --enable_training --use_cuda --cuda_version=11.4 --cuda_home="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4" --cudnn_home="C:\local\cudnn-11.4-windows-x64-v8.2.2.26\cuda" --skip_onnx_tests --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=52
|
|
msbuildPlatform: x64
|
|
isX86: false
|
|
job_name_suffix: x64_RelWithDebInfo
|
|
RunOnnxRuntimeTests: ${{ parameters.RunOnnxRuntimeTests }}
|
|
RunStaticCodeAnalysis: false
|
|
ORT_EP_NAME: CUDA
|
|
MachinePool: onnxruntime-Win2019-GPU-training
|
|
isTraining: true
|
|
|
|
- stage: dml
|
|
dependsOn: []
|
|
jobs:
|
|
- template: templates/win-gpu-ci.yml
|
|
parameters:
|
|
BuildConfig: 'RelWithDebInfo'
|
|
EnvSetupScript: setup_env.bat
|
|
buildArch: x64
|
|
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
|
|
MachinePool: onnxruntime-Win2019-GPU-dml
|
|
|