saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add on device training to the packaging pipelines (#13446)

Browse source 
### Description
enabling on device training apis in the packaging pipelines. 



### Motivation and Context
adding on device training flag so we can enable the on-device training
apis for Federated learning scenarios

Co-authored-by: Adam Louly <adamlouly@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
This commit is contained in:
Adam Louly 2022-10-25 15:03:34 -07:00 committed by GitHub
parent 7aafd86229
commit cf8bf0c141
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda116.yml
View file

@ -11,7 +11,7 @@ resources:
stages:
- template: templates/py-packaging-training-cuda-stage.yml
parameters:
build_py_parameters: --enable_training --update --build
build_py_parameters: --enable_training --update --build --enable_training_on_device
torch_version: '1.11.0'
opset_version: '15'
cuda_version: '11.6'
@ -24,7 +24,7 @@ stages:
- template: templates/py-packaging-training-cuda-stage.yml
parameters:
build_py_parameters: --enable_training --update --build
build_py_parameters: --enable_training --update --build --enable_training_on_device
torch_version: '1.11.0'
opset_version: '15'
cuda_version: '11.6'
@ -37,7 +37,7 @@ stages:
- template: templates/py-packaging-stage.yml
parameters:
build_py_parameters: --enable_training
build_py_parameters: --enable_training --enable_training_on_device
enable_linux_gpu: false
enable_ubuntu_cpu: false
enable_linux_cpu: false