saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-28 20:11:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Re-enable CUDA 12 python package test pipeline (#22370)

Browse source 
### Description
It runs after "Python-CUDA-Packaging-Pipeline" that runs on a CPU
machine that skipped all tests.
This testing pipeline is for doing the tests.
This commit is contained in:
Changming Sun 2024-10-09 20:21:27 -07:00 committed by GitHub
parent 25b1c38e87
commit dcf1e0c3b0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tools/ci_build/github/azure-pipelines/py-cuda-package-test-pipeline.yml
View file

@ -12,12 +12,13 @@ stages:
- stage: Linux_Test_GPU_x86_64_stage
dependsOn:
jobs:
- template: stages/jobs/py-linux-cuda-package-test-job.yml
- template: templates/py-packaging-linux-test-cuda.yml
parameters:
CudaVersion: '12.2'
arch: 'x86_64'
machine_pool: 'Onnxruntime-Linux-GPU'
python_wheel_suffix: '_gpu'
timeout: 480
build_id: ${{ parameters.build_id }}
project: ${{ parameters.project }}
pipeline: ${{ parameters.pipeline }}
docker_base_image: onnxruntimebuildcache.azurecr.io/internal/azureml/onnxruntime/build/cuda12_x64_ubi8_gcc12:20241008.1
trt_version: '10.4.0.26-1.cuda12.6'
cuda_version: '12.2'

2
tools/ci_build/github/azure-pipelines/templates/py-packaging-linux-test-cuda.yml
View file

@ -100,7 +100,7 @@ jobs:
inputs:
targetType: filePath
filePath: tools/ci_build/github/linux/run_python_dockertest.sh
arguments: -d GPU -c ${{parameters.cmake_build_type}} -i onnxruntimecuda118xtrt86build${{ parameters.arch }}
arguments: -d GPU -c ${{parameters.cmake_build_type}} -i onnxruntimecuda${{ replace(parameters.cuda_version, '.', '') }}xtrt86build${{ parameters.arch }}
- task: mspremier.PostBuildCleanup.PostBuildCleanup-task.PostBuildCleanup@3
displayName: 'Clean Agent Directories'