mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-27 20:02:15 +00:00
45f52987a2
### Description <!-- Describe your changes. --> ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
49 lines
1.3 KiB
YAML
49 lines
1.3 KiB
YAML
trigger: none
|
|
pr:
|
|
branches:
|
|
include:
|
|
- main
|
|
- rel-*
|
|
paths:
|
|
exclude:
|
|
- docs/**
|
|
- README.md
|
|
- CONTRIBUTING.md
|
|
- BUILD.md
|
|
- 'js/web'
|
|
- 'js/node'
|
|
- 'onnxruntime/core/providers/js'
|
|
jobs:
|
|
- template: templates/linux-ci.yml
|
|
parameters:
|
|
AgentPool : 'Onnxruntime-Linux-GPU-NC6sv3'
|
|
JobName: 'Onnxruntime_Linux_GPU_Training'
|
|
SubmoduleCheckoutMode: 'recursive'
|
|
RunDockerBuildArgs: >
|
|
-o ubuntu20.04 -d gpu
|
|
-t onnxruntime_orttraining_ortmodule_tests_image
|
|
-u
|
|
-e
|
|
-x "
|
|
--enable_training
|
|
--config $(buildConfig)
|
|
--use_cuda --cuda_version=11.8 --cuda_home=/usr/local/cuda-11.8 --cudnn_home=/usr/local/cuda-11.8
|
|
--build_wheel
|
|
--enable_nvtx_profile
|
|
--cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=70
|
|
"
|
|
DoNugetPack: 'false'
|
|
RunInjectedPipeline: 'true'
|
|
InjectedPipeline: 'orttraining-linux-gpu-ortmodule-test-ci-pipeline.yml'
|
|
DockerImageTag: 'onnxruntime_orttraining_ortmodule_tests_image'
|
|
BuildConfig: $(buildConfig)
|
|
ArtifactName: 'drop-linux'
|
|
TimeoutInMinutes: 140
|
|
# Enable unreleased onnx opsets in CI builds
|
|
# This facilitates testing the implementation for the new opsets
|
|
AllowReleasedOpsetOnly: '0'
|
|
Strategy:
|
|
maxParallel: 2
|
|
matrix:
|
|
Release:
|
|
buildConfig: Release
|