mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-27 03:11:28 +00:00
3ae5df1d18
### Description - Updates CI pipelines to use QNN SDK 2.23.0 by default. - QNN SDK adds support for int64 Cast. This allows QNN EP to support ONNX ArgMax/ArgMin/TopK operators that generate an int64 graph output. Example translation of ArgMax: - **ONNX**: input --> ArgMax --> output (int64) - **QNN**: input --> ArgMax --> Cast (int32 to int64) --> output (int64) ### Motivation and Context Update onnxruntime to use the latest QNN SDK.
80 lines
2.2 KiB
YAML
80 lines
2.2 KiB
YAML
parameters:
|
|
- name: enable_linux_cpu
|
|
displayName: 'Whether Linux CPU package is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: enable_linux_gpu
|
|
displayName: 'Whether Linux GPU package is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: enable_windows_cpu
|
|
displayName: 'Whether Windows CPU package is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: enable_windows_gpu
|
|
displayName: 'Whether Windows GPU package is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: enable_mac_cpu
|
|
displayName: 'Whether Mac CPU package is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: enable_linux_arm
|
|
displayName: 'Whether Linux ARM package is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: enable_windows_arm64_qnn
|
|
displayName: 'Whether Windows ARM64 package with QNN EP is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: enable_windows_x64_qnn
|
|
displayName: 'Whether Windows x86_64 package with QNN EP is built.'
|
|
type: boolean
|
|
default: true
|
|
|
|
- name: build_py_parameters
|
|
displayName: 'Specify extra build parameters'
|
|
type: string
|
|
default: '--use_azure'
|
|
|
|
# TODO: Now the Windows jobs use a different cmake build type. Consider to merge it.
|
|
- name: cmake_build_type
|
|
type: string
|
|
displayName: 'Linux packages cmake build type. Linux Only.'
|
|
default: 'Release'
|
|
values:
|
|
- Debug
|
|
- Release
|
|
- RelWithDebInfo
|
|
- MinSizeRel
|
|
|
|
# Only applies to QNN packages.
|
|
- name: qnn_sdk_version
|
|
type: string
|
|
displayName: 'QNN SDK version. Only for QNN packages.'
|
|
default: 2.23.0.240531
|
|
|
|
trigger: none
|
|
|
|
stages:
|
|
- template: templates/py-packaging-stage.yml
|
|
parameters:
|
|
enable_linux_gpu: ${{ parameters.enable_linux_gpu }}
|
|
enable_linux_cpu: ${{ parameters.enable_linux_cpu }}
|
|
enable_windows_cpu: ${{ parameters.enable_windows_cpu }}
|
|
enable_windows_gpu: ${{ parameters.enable_windows_gpu }}
|
|
enable_mac_cpu: ${{ parameters.enable_mac_cpu }}
|
|
enable_linux_arm: ${{ parameters.enable_linux_arm }}
|
|
enable_windows_arm64_qnn: ${{ parameters.enable_windows_arm64_qnn }}
|
|
enable_windows_x64_qnn: ${{ parameters.enable_windows_x64_qnn }}
|
|
build_py_parameters: ${{ parameters.build_py_parameters }}
|
|
cmake_build_type: ${{ parameters.cmake_build_type }}
|
|
qnn_sdk_version: ${{ parameters.qnn_sdk_version }}
|
|
publish_symbols: true
|