From 4ea54b82f9debd70e46ea0a789e7aafe05d5b983 Mon Sep 17 00:00:00 2001
From: Yi Zhang <zhanyi@microsoft.com>
Date: Wed, 3 Apr 2024 13:18:26 +0800
Subject: [PATCH] [Fix] Upload training CUDA daily wheel (#20183)

### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
---
 .../templates/py-packaging-training-cuda-stage-steps.yml        | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml b/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml
index 004c44d5f6..08f001efb6 100644
--- a/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml
+++ b/tools/ci_build/github/azure-pipelines/templates/py-packaging-training-cuda-stage-steps.yml
@@ -138,6 +138,8 @@ stages:
         value: ${{ parameters.python_version }}
       - name: Repository
         value: onnxruntimetraininggpubuild_${{ parameters.python_version }}
+      - name: UploadWheel
+        value: ${{ parameters.upload_wheel }}
     dependsOn: Build_${{ parameters.stage_name }}
     jobs:
     - job: Test_GPU