Skip to content

Commit

Permalink
Revert "Disable dml stage in windows GPU pipeline temporarily. (#18034)…
Browse files Browse the repository at this point in the history
…" (#18150)

This reverts commit 99b8dca.

### Description
<!-- Describe your changes. -->



### Motivation and Context
Restore the dml stage in windows GPU  pipeline.
Agent issue is solved by adding Feature.DisableGpuDriver in pool
properties.
  • Loading branch information
mszhanyi committed Oct 30, 2023
1 parent 8ebdd3b commit 436056d
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions tools/ci_build/github/azure-pipelines/win-gpu-ci-pipeline.yml
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,23 @@ stages:
MachinePool: onnxruntime-Win2022-GPU-T4
isTraining: true

- stage: dml
dependsOn: []
jobs:
- template: templates/jobs/win-ci-vs-2022-job.yml
parameters:
BuildConfig: 'RelWithDebInfo'
EnvSetupScript: setup_env.bat
buildArch: x64
additionalBuildFlags: --enable_pybind --use_dml --enable_wcos --use_winml
msbuildPlatform: x64
isX86: false
job_name_suffix: x64_RelWithDebInfo
RunOnnxRuntimeTests: ${{ parameters.RunOnnxRuntimeTests }}
ORT_EP_NAME: DML
WITH_CACHE: true
MachinePool: onnxruntime-Win2022-GPU-dml-A10

- stage: kernelDocumentation
dependsOn: []
jobs:
Expand Down

0 comments on commit 436056d

Please sign in to comment.