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

### 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 <[email protected]@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
microsoft · Oct 25, 2022 · cf8bf0c · cf8bf0c
1 parent 7aafd86
commit cf8bf0c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda116.yml b/tools/ci_build/github/azure-pipelines/orttraining-py-packaging-pipeline-cuda116.yml
@@ -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