Split DML test out of cuda

microsoft · Oct 26, 2024 · 7a2cd9c · 7a2cd9c
1 parent 3755562
commit 7a2cd9c
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/stages/py-win-gpu-stage.yml b/tools/ci_build/github/azure-pipelines/stages/py-win-gpu-stage.yml
@@ -71,11 +71,6 @@ stages:
         value: 'Visual Studio 17 2022'
       - name: CUDA_MODULE_LOADING
         value: 'LAZY'
-      - name: ep_name_alt
-        ${{ if eq(parameters.EP_NAME, 'directml') }}:
-          value: 'DML'
-        ${{ else }}:
-          value: 'CUDA'
 
       steps:
           - task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3
@@ -212,6 +207,6 @@ stages:
             Copy-Item -Path $(Build.sourcesDirectory)/onnxruntime/test/python/onnx_backend_test_series.py -Destination $(Agent.TempDirectory)\ort_test_data
             Copy-Item -Recurse -Path $(Build.sourcesDirectory)/onnxruntime/test/testdata -Destination $(Agent.TempDirectory)\ort_test_data
             cd $(Agent.TempDirectory)\ort_test_data
-            python onnx_backend_test_series.py --devices $(ep_name_alt)
+            python onnx_backend_test_series.py --devices DML
           workingDirectory: '$(Build.sourcesDirectory)'
           displayName: 'Run Python Tests'