Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,10 @@ stages:
displayName: 'Download Pipeline Artifact - Win GPU 3.13'
artifact: 'win_gpu_wheel_3.13'
patterns: '*.whl'

- download: build
displayName: 'Download Pipeline Artifact - Win GPU 3.14'
artifact: 'win_gpu_wheel_3.14'
patterns: '*.whl'
- script: find $(Pipeline.Workspace) -name \*win_amd64.whl -exec mv {} $(Pipeline.Workspace)/build/onnxruntime_gpu \;
displayName: 'Merge files together'

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,10 @@ extends:
EP_BUILD_FLAGS: --use_dml
EP_NAME: directml
cmake_build_type: ${{ parameters.cmake_build_type }}
- template: stages/py-win-gpu-stage.yml
parameters:
MACHINE_POOL: 'onnxruntime-Win2022-GPU-dml-A10'
PYTHON_VERSION: '3.14'
EP_BUILD_FLAGS: --use_dml
EP_NAME: directml
cmake_build_type: ${{ parameters.cmake_build_type }}
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,8 @@ stages:
PythonVersion: '3.12'
Python313_x64:
PythonVersion: '3.13'
Python314_x64:
PythonVersion: '3.14'
variables:
OnnxRuntimeBuildDirectory: '$(Build.BinariesDirectory)'
ExtraParam: ${{ parameters.build_py_parameters }}
Expand Down Expand Up @@ -138,7 +140,7 @@ stages:
parameters:
${{if eq(variables['Build.SourceBranch'], 'refs/heads/main')}}:
symbolExpiryTime: 60
includePublicSymbolServer: true
includePublicSymbolServer: true
symbolsArtifactName: onnxruntime_cpu_win_x64_$(PythonVersion)
symbolsVersion: $(Build.BuildId)
symbolProject: 'ONNX Runtime'
Expand Down Expand Up @@ -204,14 +206,21 @@ stages:
extra_build_arg: ${{ parameters.build_py_parameters }}
cmake_build_type: ${{ parameters.cmake_build_type }}
python_version: '3.12'

- template: ../templates/py-macos.yml
parameters:
arch: 'arm64'
extra_build_arg: ${{ parameters.build_py_parameters }}
cmake_build_type: ${{ parameters.cmake_build_type }}
python_version: '3.13'

- template: ../templates/py-macos.yml
parameters:
arch: 'arm64'
extra_build_arg: ${{ parameters.build_py_parameters }}
cmake_build_type: ${{ parameters.cmake_build_type }}
python_version: '3.14'

- ${{ if eq(parameters.enable_linux_arm, true) }}:
- stage: Python_Packaging_Linux_ARM
dependsOn: []
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ parameters:
- '3.11'
- '3.12'
- '3.13'
- '3.14'

stages:
- ${{ each python_version in parameters.PythonVersions }}:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,8 @@ jobs:
PythonVersion: '3.12'
Python313:
PythonVersion: '3.13'
Python314:
PythonVersion: '3.14'
steps:
- checkout: none

Expand All @@ -58,4 +60,3 @@ jobs:
python3 -c "import onnxruntime as ort; print(ort.__version__)"
workingDirectory: $(Pipeline.Workspace)/build/onnxruntime-${{ parameters.arch }}-${{ parameters.ep }}
displayName: Test Package Installation

Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ parameters:
default: '12.8'
values:
- 12.8
- 13.0

# TODO: Ideally it should fetch information from the build that triggers it
- name: cmake_build_type
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,8 @@ jobs:
PythonVersion: '3.12'
Python313_x64:
PythonVersion: '3.13'
Python314_x64:
PythonVersion: '3.14'
variables:
GRADLE_OPTS: '-Dorg.gradle.daemon=false'
VSGenerator: 'Visual Studio 17 2022'
Expand Down Expand Up @@ -113,5 +115,3 @@ jobs:
inputs:
artifactName: onnxruntime_qnn_arm64ec_$(PythonVersion)
targetPath: '$(Build.ArtifactStagingDirectory)'


Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,8 @@ jobs:
PythonVersion: '3.12'
Python313_x64:
PythonVersion: '3.13'
Python314_x64:
PythonVersion: '3.14'
variables:
GRADLE_OPTS: '-Dorg.gradle.daemon=false'
VSGenerator: 'Visual Studio 17 2022'
Expand All @@ -51,9 +53,9 @@ jobs:
parameters:
host_cpu_arch: 'x64'
python_version: $(PythonVersion)

- script: python -m pip install -r $(Build.SourcesDirectory)\tools\ci_build\github\windows\python\requirements.txt

- template: set-nightly-build-option-variable-step.yml

- template: jobs/download_win_qnn_sdk.yml
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,10 @@ steps:
variables = {
"PythonManylinuxDir": "/opt/python/cp313-cp313"
}
elif version == "3.14":
variables = {
"PythonManylinuxDir": "/opt/python/cp314-cp314"
}
else:
raise ValueError("Unsupported Python version: '{}'".format(version))

Expand Down
2 changes: 1 addition & 1 deletion tools/ci_build/github/azure-pipelines/templates/win-ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -77,8 +77,8 @@ parameters:
type: string
default: '12.8'
values:
- 13.0
- 12.8
- 13.0

- name: SpecificArtifact
displayName: Use Specific Artifact
Expand Down
4 changes: 2 additions & 2 deletions tools/ci_build/github/linux/build_linux_python_package.sh
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@
PYTHON_EXES=(
"/opt/python/cp311-cp311/bin/python3.11"
"/opt/python/cp313-cp313/bin/python3.13"
"/opt/python/cp313-cp313t/bin/python3.13t"
"/opt/python/cp312-cp312/bin/python3.12"
"/opt/python/cp314-cp314/bin/python3.14"
"/opt/python/cp312-cp312/bin/python3.12"
)
while getopts "d:p:x:c:e" parameter_Option
do case "${parameter_Option}"
Expand Down Expand Up @@ -67,7 +67,7 @@
fi

if [ "$BUILD_DEVICE" == "GPU" ]; then
SHORT_CUDA_VERSION=$(echo $CUDA_VERSION | sed 's/\([[:digit:]]\+\.[[:digit:]]\+\)\.[[:digit:]]\+/\1/')

Check warning on line 70 in tools/ci_build/github/linux/build_linux_python_package.sh

View workflow job for this annotation

GitHub Actions / Optional Lint

[shellcheck] reported by reviewdog 🐶 Double quote to prevent globbing and word splitting. Raw Output: ./tools/ci_build/github/linux/build_linux_python_package.sh:70:31: info: Double quote to prevent globbing and word splitting. (ShellCheck.SC2086)

Check warning on line 70 in tools/ci_build/github/linux/build_linux_python_package.sh

View workflow job for this annotation

GitHub Actions / Optional Lint

[shellcheck] reported by reviewdog 🐶 See if you can use ${variable//search/replace} instead. Raw Output: ./tools/ci_build/github/linux/build_linux_python_package.sh:70:26: info: See if you can use ${variable//search/replace} instead. (ShellCheck.SC2001)
#Enable CUDA and TRT EPs.
BUILD_ARGS+=("--use_cuda" "--use_tensorrt" "--cuda_version=$SHORT_CUDA_VERSION" "--tensorrt_home=/usr" "--cuda_home=/usr/local/cuda-$SHORT_CUDA_VERSION" "--cudnn_home=/usr/local/cuda-$SHORT_CUDA_VERSION" "--nvcc_threads=1" "--cmake_extra_defines" "CMAKE_CUDA_ARCHITECTURES=60-real;70-real;75-real;80-real;86-real;90a-real;90-virtual" "onnxruntime_USE_FPA_INTB_GEMM=OFF")
fi
Expand Down
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
#!/bin/bash
set -e -x
pushd .
PYTHON_EXES=("/opt/python/cp311-cp311/bin/python3.11" "/opt/python/cp312-cp312/bin/python3.12" "/opt/python/cp313-cp313/bin/python3.13" "/opt/python/cp313-cp313t/bin/python3.13")
PYTHON_EXES=("/opt/python/cp311-cp311/bin/python3.11" "/opt/python/cp312-cp312/bin/python3.12" "/opt/python/cp313-cp313/bin/python3.13" "/opt/python/cp314-cp314/bin/python3.14")
CURRENT_DIR=$(pwd)
if ! [ -x "$(command -v protoc)" ]; then
$CURRENT_DIR/install_protobuf.sh

Check warning on line 7 in tools/ci_build/github/linux/docker/inference/aarch64/python/cpu/scripts/install_deps.sh

View workflow job for this annotation

GitHub Actions / Optional Lint

[shellcheck] reported by reviewdog 🐶 Double quote to prevent globbing and word splitting. Raw Output: ./tools/ci_build/github/linux/docker/inference/aarch64/python/cpu/scripts/install_deps.sh:7:3: info: Double quote to prevent globbing and word splitting. (ShellCheck.SC2086)
fi
popd
export ONNX_ML=1
Expand All @@ -13,4 +13,4 @@
for PYTHON_EXE in "${PYTHON_EXES[@]}"
do
${PYTHON_EXE} -m pip install -r requirements.txt
done
done
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,12 @@ mypy
pytest
setuptools>=68.2.2
wheel
onnx==1.19.1
protobuf==4.25.8
sympy==1.14
flatbuffers

# Use onnx for Python < 3.14
onnx==1.19.1; python_version < "3.14"

# Use onnx_weekly for Python >= 3.14
onnx_weekly; python_version >= "3.14"
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ PARENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." &> /dev/null && pwd)"
source "$PARENT_DIR/install_dotnet.sh"

if [ ! -d "/opt/conda/bin" ]; then
PYTHON_EXES=("/opt/python/cp311-cp311/bin/python3.11" "/opt/python/cp312-cp312/bin/python3.12")
PYTHON_EXES=("/opt/python/cp311-cp311/bin/python3.11" "/opt/python/cp312-cp312/bin/python3.12" "/opt/python/cp313-cp313/bin/python3.13" "/opt/python/cp314-cp314/bin/python3.14")
else
PYTHON_EXES=("/opt/conda/bin/python")
fi
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,4 +8,4 @@ protobuf==4.25.1
sympy==1.14
flatbuffers
neural-compressor>=2.2.1
triton==3.2.0
triton
Loading