mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-04 23:59:56 +00:00
ee543a47f6
* upgrade cuda version on ci pipelines * keeping folder name same * keeping folder name same * setting manual seed for primitive test case * resolving comments * changing atol and rtrol only for test case Co-authored-by: Adam Louly <adamlouly@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
36 lines
1.1 KiB
YAML
36 lines
1.1 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.6 --cuda_home=/usr/local/cuda-11.6 --cudnn_home=/usr/local/cuda-11.6
|
|
--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
|