From 652950e586b06602cf6fee51c95e661ed78b4847 Mon Sep 17 00:00:00 2001 From: Yi Zhang Date: Fri, 25 Oct 2024 07:34:12 +0800 Subject: [PATCH] add --use-winml --- .../stages/nuget-win-cuda-packaging-stage.yml | 4 ++-- tools/ci_build/github/azure-pipelines/templates/win-ci.yml | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tools/ci_build/github/azure-pipelines/stages/nuget-win-cuda-packaging-stage.yml b/tools/ci_build/github/azure-pipelines/stages/nuget-win-cuda-packaging-stage.yml index be9abc5640cf4..99b972fac2455 100644 --- a/tools/ci_build/github/azure-pipelines/stages/nuget-win-cuda-packaging-stage.yml +++ b/tools/ci_build/github/azure-pipelines/stages/nuget-win-cuda-packaging-stage.yml @@ -34,7 +34,7 @@ parameters: displayName: Specific Artifact's BuildId type: string default: '0' - + - name: buildJava type: boolean @@ -50,7 +50,7 @@ stages: msbuildPlatform: x64 packageName: x64-cuda CudaVersion: ${{ parameters.CudaVersion }} - buildparameter: --use_cuda --cuda_home=${{ parameters.win_cuda_home }} --enable_wcos --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=60;61;70;75;80" --use_dml --build_csharp --parallel + buildparameter: --use_cuda --cuda_home=${{ parameters.win_cuda_home }} --enable_wcos --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=60;61;70;75;80" --use_dml --use_winml --build_csharp --parallel runTests: ${{ parameters.RunOnnxRuntimeTests }} buildJava: ${{ parameters.buildJava }} java_artifact_id: onnxruntime_gpu diff --git a/tools/ci_build/github/azure-pipelines/templates/win-ci.yml b/tools/ci_build/github/azure-pipelines/templates/win-ci.yml index aa315002bb0b3..dd3ee17184f15 100644 --- a/tools/ci_build/github/azure-pipelines/templates/win-ci.yml +++ b/tools/ci_build/github/azure-pipelines/templates/win-ci.yml @@ -319,9 +319,9 @@ stages: - job: Windows_Packaging_${{ parameters.stage_name_suffix }}_Testing workspace: clean: all - pool: zhanyi_test_A100_pool - # name: ${{ parameters.ort_build_pool_name }} - # demands: ImageVersionOverride -equals 248.0.0 + pool: + name: ${{ parameters.ort_build_pool_name }} + demands: ImageVersionOverride -equals 248.0.0 timeoutInMinutes: 180 steps: - task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3