fix type for py-win-gpu.yml

microsoft · Nov 15, 2023 · 79c2c58 · 79c2c58
1 parent 7058474
commit 79c2c58
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/templates/py-win-gpu.yml b/tools/ci_build/github/azure-pipelines/templates/py-win-gpu.yml
@@ -71,16 +71,16 @@ jobs:
         - template: jobs/set-winenv.yml
           parameters:
             EnvSetupScript: ${{ parameters.ENV_SETUP_SCRIPT }}
-            ${{ if contains(parameters.buildparameter, 'use_cuda') }}:
+            ${{ if contains(parameters.EP_BUILD_FLAGS, 'use_cuda') }}:
               DownloadCUDA: true
 
       - ${{ if eq(parameters.ENV_SETUP_SCRIPT, '') }}:
         - template: jobs/download_win_gpu_library.yml
           parameters:
             CudaVersion: ${{ parameters.CudaVersion }}
-            ${{ if contains(parameters.buildparameter, 'use_cuda') }}:
+            ${{ if contains(parameters.EP_BUILD_FLAGS, 'use_cuda') }}:
               DownloadCUDA: true
-            ${{ if contains(parameters.buildparameter, 'use_tensorrt') }}:
+            ${{ if contains(parameters.EP_BUILD_FLAGS, 'use_tensorrt') }}:
               DownloadCUDA: true
               DownloadTRT: true
       - task: PythonScript@0