Skip to content

Enable CUDA Python Test #22717

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Jian Chen (jchen351) merged 6 commits into from
Nov 6, 2024
Merged

Enable CUDA Python Test #22717

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
b8b98ea
Enable Win CUDA python test
jchen351 Nov 4, 2024
bcd173b
Enable Win CUDA python test
jchen351 Nov 4, 2024
015b32b
exclude failed cuda 12 tests
jchen351 Nov 5, 2024
bc4e825
Adding verbose to run onnx_backend_test_series.py
jchen351 Nov 5, 2024
099d7dd
remove CudaVersion: ${{ parameters.cuda_version }}
jchen351 Nov 5, 2024
3369d40
remove 'setup env'
jchen351 Nov 5, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions onnxruntime/test/testdata/onnx_backend_test_series_filters.jsonc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -306,6 +306,11 @@
"^test_qlinearmatmul_3D_int8_float32_cuda",
"^test_qlinearmatmul_3D_uint8_float16_cuda",
"^test_qlinearmatmul_3D_uint8_float32_cuda",
// Tests that failed on CUDA 12.2.
"^test_Conv3d_dilated_cuda",
"^test_Conv3d_dilated_strided_cuda",
Comment thread
jchen351 marked this conversation as resolved.
"^test_Conv3d_stride_cuda",
"^test_Conv3d_stride_padding_cuda",
// Size(21) from ONNX 1.16.0 is not implemented in cuda.
"^test_size_cuda",
"^test_size_example_cuda",
Expand Down
1 change: 1 addition & 0 deletions tools/ci_build/github/azure-pipelines/py-cuda-packaging-pipeline.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ parameters:
- RelWithDebInfo
- MinSizeRel


extends:
# The pipeline extends the 1ES PT which will inject different SDL and compliance tasks.
# For non-production pipelines, use "Unofficial" as defined below.
Expand Down
1 change: 0 additions & 1 deletion tools/ci_build/github/azure-pipelines/stages/py-gpu-packaging-stage.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,5 @@ stages:
MACHINE_POOL: 'onnxruntime-Win2022-GPU-dml-A10'
PYTHON_VERSION: ${{ python_version }}
EP_BUILD_FLAGS: --use_dml --cmake_extra_defines CMAKE_SYSTEM_VERSION=10.0.18362.0 --enable_wcos
ENV_SETUP_SCRIPT: setup_env.bat
EP_NAME: directml
cmake_build_type: ${{ parameters.cmake_build_type }}
32 changes: 13 additions & 19 deletions tools/ci_build/github/azure-pipelines/stages/py-win-gpu-stage.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,10 +12,6 @@ parameters:
- name: EP_BUILD_FLAGS
type: string

- name: ENV_SETUP_SCRIPT
type: string
default: ''

- name: BUILD_PY_PARAMETERS
displayName: >
Extra parameters to pass to build.py. Don't put newlines in here.
Expand Down Expand Up @@ -111,21 +107,12 @@ stages:

- template: ../templates/download-deps.yml

- ${{ if ne(parameters.ENV_SETUP_SCRIPT, '') }}:
- template: ../templates/jobs/set-winenv.yml
parameters:
EnvSetupScript: ${{ parameters.ENV_SETUP_SCRIPT }}
${{ if or(contains(parameters.EP_BUILD_FLAGS, 'use_cuda'), eq(parameters.use_tensorrt, true)) }}:
DownloadCUDA: true
DownloadTRT: ${{ parameters.use_tensorrt }}

- ${{ if eq(parameters.ENV_SETUP_SCRIPT, '') }}:
- template: ../templates/jobs/download_win_gpu_library.yml
parameters:
CudaVersion: ${{ parameters.CudaVersion }}
${{ if or(contains(parameters.EP_BUILD_FLAGS, 'use_cuda'), eq(parameters.use_tensorrt, true)) }}:
DownloadCUDA: true
DownloadTRT: ${{ parameters.use_tensorrt }}
- template: ../templates/jobs/download_win_gpu_library.yml
parameters:
CudaVersion: ${{ parameters.CudaVersion }}
${{ if or(contains(parameters.EP_BUILD_FLAGS, 'use_cuda'), eq(parameters.use_tensorrt, true)) }}:
DownloadCUDA: true
DownloadTRT: ${{ parameters.use_tensorrt }}

- task: PythonScript@0
displayName: 'Update deps.txt'
Expand Down Expand Up @@ -211,6 +198,13 @@ stages:
StepName: 'Download Pipeline Artifact - Windows GPU Build'
TargetPath: '$(Build.ArtifactStagingDirectory)'

- template: ../templates/jobs/download_win_gpu_library.yml
parameters:
CudaVersion: ${{ parameters.CudaVersion }}
${{ if or(contains(parameters.EP_BUILD_FLAGS, 'use_cuda'), eq(parameters.use_tensorrt, true)) }}:
DownloadCUDA: true
DownloadTRT: ${{ parameters.use_tensorrt }}

- task: PowerShell@2
displayName: 'Install ONNX'
inputs:
Expand Down