mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-28 03:20:58 +00:00
4af593a722
1. Add python 3.13 to our python packaging pipelines 2. Because numpy 2.0.0 doesn't support thread free python, this PR also upgrades numpy to the latest 3. Delete some unused files.
88 lines
2.9 KiB
YAML
88 lines
2.9 KiB
YAML
resources:
|
|
pipelines:
|
|
- pipeline: build
|
|
source: 'Python packaging pipeline'
|
|
trigger: true
|
|
branch: main # branch to pick the artifact, Used only for manual triggered pipeline runs for testing the pipeline itself
|
|
|
|
stages:
|
|
- stage: Linux_Test_CPU_x86_64_stage
|
|
jobs:
|
|
- template: templates/py-packaging-linux-test-cpu.yml
|
|
parameters:
|
|
arch: 'x86_64'
|
|
machine_pool: 'onnxruntime-Ubuntu2204-AMD-CPU'
|
|
|
|
|
|
- stage: Linux_Test_CPU_aarch64_stage
|
|
dependsOn: []
|
|
jobs:
|
|
- template: templates/py-packaging-linux-test-cpu.yml
|
|
parameters:
|
|
arch: 'aarch64'
|
|
machine_pool: 'onnxruntime-linux-ARM64-CPU-2019'
|
|
|
|
- stage: Packages_Somking_Test
|
|
dependsOn: []
|
|
jobs:
|
|
- template: templates/py-package-smoking-test.yml
|
|
parameters:
|
|
job_name: Test_MAC_Wheels
|
|
machine_pool:
|
|
vmImage: 'macOS-13'
|
|
itemPattern: '*/*mac*x86_64.whl'
|
|
- template: templates/py-package-smoking-test.yml
|
|
parameters:
|
|
job_name: Test_LINUX_x86_64_Wheels
|
|
itemPattern: '*/*manylinux*x86_64.whl'
|
|
machine_pool:
|
|
name: 'onnxruntime-Ubuntu2204-AMD-CPU'
|
|
|
|
# ****The following Stage depend on all previous tags. ***
|
|
|
|
# GPU resources are very limited,
|
|
# To utilize gpu resource more efficiently, run GPU job only after all cpus jobs succeed
|
|
- stage: Linux_Test_GPU_x86_64_stage
|
|
dependsOn:
|
|
- Linux_Test_CPU_x86_64_stage
|
|
- Linux_Test_CPU_aarch64_stage
|
|
- Packages_Somking_Test
|
|
jobs:
|
|
- template: templates/py-packaging-linux-test-cuda.yml
|
|
parameters:
|
|
arch: 'x86_64'
|
|
machine_pool: 'Onnxruntime-Linux-GPU'
|
|
python_wheel_suffix: '_gpu'
|
|
timeout: 480
|
|
docker_base_image: onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cuda11_x64_almalinux8_gcc11:20241010.2
|
|
trt_version: '10.4.0.26-1.cuda11.8'
|
|
cuda_version: '11.8'
|
|
|
|
|
|
# if final job not extecuted, it will not run nightlly build
|
|
- stage: Final
|
|
dependsOn:
|
|
- Linux_Test_GPU_x86_64_stage
|
|
jobs:
|
|
- job: Final
|
|
# Run this step only if all previous steps are succeeded and (this build was triggered by a resource trigger or it was triggered by another build).
|
|
condition: and(succeeded(), eq(variables['Build.Reason'], 'ResourceTrigger'))
|
|
timeoutInMinutes: 60
|
|
variables:
|
|
skipComponentGovernanceDetection: true
|
|
workspace:
|
|
clean: all
|
|
pool: Onnxruntime-Win-CPU-2022
|
|
steps:
|
|
- task: PowerShell@2
|
|
displayName: 'Add Build Tag'
|
|
env:
|
|
SYSTEM_ACCESSTOKEN: $(System.AccessToken)
|
|
inputs:
|
|
targetType: inline
|
|
script: |
|
|
$headers = New-Object "System.Collections.Generic.Dictionary[[String],[String]]"
|
|
$headers.Add("Authorization", "Bearer $env:SYSTEM_ACCESSTOKEN")
|
|
$headers.Add("Content-Type", "application/json")
|
|
$uri = "https://dev.azure.com/aiinfra/Lotus/_apis/build/builds/$(resources.pipeline.build.runID)/tags/test%20pass?api-version=6.0"
|
|
Invoke-RestMethod -Uri $uri -Headers $headers -Method PUT
|