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 index 50cf87c91e51..ca6eb1da084e 100644 --- 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 @@ -72,7 +72,7 @@ stages: - name: CUDA_MODULE_LOADING value: 'LAZY' - name: ep_name_alt - (parameters.EP_NAME, 'directml') }}: + ${{ if eq(parameters.EP_NAME, 'directml') }}: value: 'DML' ${{ else }}: value: 'CUDA'