saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-02 23:39:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert "Disable dml stage in windows GPU pipeline temporarily. (#18034)" (#18150) (#18170)

Browse source 
This reverts commit 99b8dcaae2.


### Motivation and Context
Restore the dml stage in windows GPU  pipeline.
Agent issue is solved by adding Feature.DisableGpuDriver in pool
properties.

---------

Co-authored-by: Yi Zhang <zhanyi@microsoft.com>
This commit is contained in:
Changming Sun 2023-10-30 21:57:00 -07:00 committed by GitHub
parent 99b0f626ac
commit 6ae7c51a2f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tools/ci_build/github/azure-pipelines/win-gpu-ci-pipeline.yml
View file

@ -72,6 +72,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:

2
winml/test/model/skip_model_tests.h
View file

@ -146,6 +146,8 @@ std::unordered_map<std::string, std::string> disabledTests({
"Bug 31005780: Result of fp16_test_tiny_yolov2_opset7 and fp16_coreml_FNS_Candy_opset7 models on DirectML aren't as accurate as on CPU https://microsoft.visualstudio.com/OS/_workitems/edit/31005780"},
{ "mlperf_ssd_mobilenet_300_opset10_GPU",
"Bug 31005624: mlperf_ssd_mobilenet_300 opset 10 model fails to evaluate in DirectML https://microsoft.visualstudio.com/OS/_workitems/edit/31005624" },
{ "mlperf_ssd_resnet34_1200_opset10_GPU",
"Bug 31005624: mlperf_ssd_resnet34_1200_opset10_GPU opset 10 model fails to evaluate in DirectML https://microsoft.visualstudio.com/OS/_workitems/edit/31005624" },
});
/*