mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-30 20:18:08 +00:00
Adding new Python package testing pipeline for Cuda Alt (#22584)
### Description <!-- Describe your changes. --> ### 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. -->
This commit is contained in:
parent
15556c492d
commit
3fe7aa3b59
3 changed files with 28 additions and 18 deletions
|
|
@ -0,0 +1,24 @@
|
|||
resources:
|
||||
pipelines:
|
||||
- pipeline: build
|
||||
source: 'Python CUDA ALT Packaging Pipeline'
|
||||
trigger: true
|
||||
branch: main # branch to pick the artifact, Used only for manual triggered pipeline runs for testing the pipeline itself
|
||||
|
||||
stages:
|
||||
# ****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_CUDA_Alt_x86_64_stage
|
||||
dependsOn:
|
||||
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:20241020.1
|
||||
trt_version: '10.4.0.26-1.cuda11.8'
|
||||
cuda_version: '11.8'
|
||||
|
||||
|
|
@ -9,7 +9,7 @@ stages:
|
|||
# ****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
|
||||
- stage: Linux_Test_CUDA_x86_64_stage
|
||||
dependsOn:
|
||||
jobs:
|
||||
- template: templates/py-packaging-linux-test-cuda.yml
|
||||
|
|
|
|||
|
|
@ -42,27 +42,13 @@ stages:
|
|||
|
||||
# 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:20241020.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
|
||||
- Linux_Test_CPU_x86_64_stage
|
||||
- Linux_Test_CPU_aarch64_stage
|
||||
- Packages_Somking_Test
|
||||
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).
|
||||
|
|
|
|||
Loading…
Reference in a new issue