mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-01 23:30:35 +00:00
bb76ead96c
### Description <!-- Describe your changes. --> * promote trt version to 10.2.0.19 * EP_Perf CI: clean config of legacy TRT<8.6, promote test env to trt10.2-cu118/cu125 * skip two tests as Float8/BF16 are supported by TRT>10.0 but TRT CIs are not hardware-compatible on these: ``` 1: [ FAILED ] 2 tests, listed below: 1: [ FAILED ] IsInfTest.test_isinf_bfloat16 1: [ FAILED ] IsInfTest.test_Float8E4M3FN ``` ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
88 lines
3 KiB
YAML
88 lines
3 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-latest'
|
|
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:20240531.1
|
|
trt_version: '10.2.0.19-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
|