From cf8bf0c1415e8f454776435549c50ef4b2ec1bc2 Mon Sep 17 00:00:00 2001 From: Adam Louly Date: Tue, 25 Oct 2022 15:03:34 -0700 Subject: [PATCH] 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 --- .../orttraining-py-packaging-pipeline-cuda116.yml | 6 +++--- 1 file changed, 3 insertions(+), 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 index 2befd8838f..253c4adc9c 100644 --- 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