mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-06 00:03:22 +00:00
37 lines
1.2 KiB
YAML
37 lines
1.2 KiB
YAML
trigger: none
|
|
|
|
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
|
|
-e
|
|
-x "
|
|
--enable_training
|
|
--enable_training_torch_interop
|
|
--config $(buildConfig)
|
|
--use_cuda --cuda_version=11.3 --cuda_home=/usr/local/cuda-11.3 --cudnn_home=/usr/local/cuda-11.3
|
|
--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
|
|
|