mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-30 20:18:08 +00:00
f9e623e4d1
To include a bug fix: https://gitlab.kitware.com/cmake/cmake/-/merge_requests/9890 Discussion: https://discourse.cmake.org/t/cmake-incorrectly-links-to-nvrtc-builtins/12723/4 This bug fix should be included in our upcoming release, because right now our GPU package depends on “libnvrtc-builtins.so.12.2" which has a hardcoded CUDA version: 12.2. The minor CUDA version should not be there.
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:20241015.1
|
|
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
|