Update build-perf-test-binaries-pipeline.yml

microsoft · Oct 21, 2024 · 4fb8c84 · 4fb8c84
1 parent c7138a2
commit 4fb8c84
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/build-perf-test-binaries-pipeline.yml b/tools/ci_build/github/azure-pipelines/build-perf-test-binaries-pipeline.yml
@@ -34,11 +34,8 @@ stages:
 # build Python packages
 # Linux GPU only
 - ${{ if parameters.BuildPythonPackages }}:
-  - template: templates/py-packaging-stage.yml
+  - template: stages/py-cuda-packaging-stage.yml
     parameters:
       enable_linux_gpu: true
-      enable_linux_cpu: false
-      enable_windows_cpu: false
       enable_windows_gpu: false
-      enable_mac_cpu: false
-      enable_linux_arm: false
+      cuda_version: 12.2