Skip to content

Commit 0409ba6

Browse files
authored
Build onnxruntime with caller-framework WinAI for FoundryLocal (#26915)
Aligns onnxruntime-foundry `caller-framework` build flag with the windows-ml build flag so telemetry can work through the same infrastructure.
1 parent 34a0b15 commit 0409ba6

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tools/ci_build/github/azure-pipelines/custom-nuget-packaging-pipeline.yml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ parameters:
3131
- name: PackageName
3232
displayName: What is the package name? Override using an environment variable CustomPackageName.
3333
type: string
34-
default: 'Microsoft.ML.OnnxRuntime.Flamingo'
34+
default: 'Microsoft.ML.OnnxRuntime.Foundry'
3535

3636
variables:
3737
- template: templates/common-variables.yml
@@ -107,7 +107,7 @@ extends:
107107
msbuildPlatform: x64
108108
packageName: x64-cuda
109109
CudaVersion: ${{ parameters.CudaVersion }}
110-
buildparameter: --use_cuda --cuda_home=${{ variables.win_cuda_home }} --enable_onnx_tests --use_webgpu --parallel 4 --nvcc_threads 1 --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=${{ variables.CmakeCudaArchitectures }}"
110+
buildparameter: --use_cuda --cuda_home=${{ variables.win_cuda_home }} --enable_onnx_tests --use_webgpu --parallel 4 --nvcc_threads 1 --caller_framework WinAI --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=${{ variables.CmakeCudaArchitectures }}"
111111
runTests: false
112112
buildJava: false
113113
java_artifact_id: onnxruntime_gpu
@@ -127,6 +127,7 @@ extends:
127127
build_config: 'RelWithDebInfo'
128128
PublishArchive: true
129129
PublishNugetToFeed: false
130+
AdditionalBuildArgs: '--caller_framework WinAI'
130131

131132
- template: templates/mac-cpu-packaging-pipeline.yml
132133
parameters:

tools/ci_build/github/azure-pipelines/templates/qnn-ep-win.yml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ parameters:
88
StageName: 'OnnxRuntime_QNN_Nuget_Win_x64'
99
PublishArchive: false
1010
PublishNugetToFeed: true
11+
AdditionalBuildArgs: ''
1112

1213
stages:
1314
- stage: ${{ parameters.StageName }}
@@ -47,7 +48,7 @@ stages:
4748
variables:
4849
OrtPackageId: ${{ parameters.OrtNugetPackageId }}
4950
ReleaseVersionSuffix: $[stageDependencies.Setup.Set_Variables.outputs['Set_Release_Version_Suffix.ReleaseVersionSuffix']]
50-
commonBuildArgs: '--skip_submodule_sync --build_shared_lib --client_package_build --cmake_generator "Visual Studio 17 2022" --config ${{ parameters.build_config }} --parallel --use_vcpkg --use_vcpkg_ms_internal_asset_cache --use_binskim_compliant_compile_flags '
51+
commonBuildArgs: '--skip_submodule_sync --build_shared_lib --client_package_build --cmake_generator "Visual Studio 17 2022" --config ${{ parameters.build_config }} --parallel --use_vcpkg --use_vcpkg_ms_internal_asset_cache --use_binskim_compliant_compile_flags ${{ parameters.AdditionalBuildArgs}}'
5152

5253
steps:
5354
- template: set-version-number-variables-step.yml
@@ -65,7 +66,7 @@ stages:
6566
inputs:
6667
scriptPath: '$(Build.SourcesDirectory)\tools\ci_build\build.py'
6768
arguments: ' --arm64 --buildasx --build_dir $(Build.BinariesDirectory)\arm64x --use_qnn --qnn_home $(QnnSDKRootDir) $(commonBuildArgs)'
68-
69+
6970
- task: PythonScript@0
7071
displayName: 'Build arm64ecx project - the real arm64x'
7172
inputs:

0 commit comments

Comments
 (0)