${{ if or(contains(parameters.EP_BUILD_FLAGS, 'use_cuda'), contains(…

…parameters.EP_BUILD_FLAGS, 'use_tensorrt')) }}:
microsoft · Nov 15, 2023 · 38fa4ab · 38fa4ab
1 parent 4d7846f
commit 38fa4ab
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
@@ -76,18 +76,18 @@ jobs:
         - template: jobs/set-winenv.yml
           parameters:
             EnvSetupScript: ${{ parameters.ENV_SETUP_SCRIPT }}
-            ${{ if contains(parameters.EP_BUILD_FLAGS, 'use_cuda') }}:
+            ${{ if or(contains(parameters.EP_BUILD_FLAGS, 'use_cuda'), contains(parameters.EP_BUILD_FLAGS, 'use_tensorrt')) }}:
               DownloadCUDA: true
 
       - ${{ if eq(parameters.ENV_SETUP_SCRIPT, '') }}:
         - template: jobs/download_win_gpu_library.yml
           parameters:
             CudaVersion: ${{ parameters.CudaVersion }}
-            ${{ if contains(parameters.EP_BUILD_FLAGS, 'use_cuda') }}:
+            ${{ if or(contains(parameters.EP_BUILD_FLAGS, 'use_cuda'), contains(parameters.EP_BUILD_FLAGS, 'use_tensorrt')) }}:
               DownloadCUDA: true
             ${{ if contains(parameters.EP_BUILD_FLAGS, 'use_tensorrt') }}:
-              DownloadCUDA: true
               DownloadTRT: true
+
       - task: PythonScript@0
         displayName: 'Update deps.txt'
         inputs: