mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-20 19:12:24 +00:00
1fc9c4823d
### Description Enable coremltools for Linux build. In order to do this, I did: 1. Add uuid-devel to the Linux images and regenerate them. 2. Patch the coremltools code a little bit to add some missing header files. ### Motivation and Context To make the code simpler. Later on I will create another PR to remove the COREML_ENABLE_MLPROGRAM C/C++ macro. Also, after this PR I will bring more changes to onnxruntime_provider_coreml.cmake to make it work with vcpkg.
57 lines
No EOL
2.1 KiB
YAML
57 lines
No EOL
2.1 KiB
YAML
resources:
|
|
pipelines:
|
|
- pipeline: build
|
|
source: 'Python-CUDA-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_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/cuda12_x64_ubi8_gcc12:20250124.1
|
|
cuda_version: '12.2'
|
|
|
|
- stage: Republish_Wheels
|
|
dependsOn:
|
|
jobs:
|
|
- job: Python_Publishing_GPU
|
|
pool: 'onnxruntime-Ubuntu2204-AMD-CPU'
|
|
steps:
|
|
- checkout: none
|
|
- download: build
|
|
displayName: 'Download Pipeline Artifact - onnxruntime_gpu'
|
|
artifact: 'onnxruntime_gpu'
|
|
patterns: '*.whl'
|
|
- download: build
|
|
displayName: 'Download Pipeline Artifact - Win GPU 3.10'
|
|
artifact: 'win_gpu_wheel_3.10'
|
|
patterns: '*.whl'
|
|
- download: build
|
|
displayName: 'Download Pipeline Artifact - Win GPU 3.11'
|
|
artifact: 'win_gpu_wheel_3.11'
|
|
patterns: '*.whl'
|
|
- download: build
|
|
displayName: 'Download Pipeline Artifact - Win GPU 3.12'
|
|
artifact: 'win_gpu_wheel_3.12'
|
|
patterns: '*.whl'
|
|
- download: build
|
|
displayName: 'Download Pipeline Artifact - Win GPU 3.13'
|
|
artifact: 'win_gpu_wheel_3.13'
|
|
patterns: '*.whl'
|
|
|
|
- script: find $(Pipeline.Workspace) -name \*win_amd64.whl -exec mv {} $(Pipeline.Workspace)/build/onnxruntime_gpu \;
|
|
displayName: 'Merge files together'
|
|
|
|
- publish: $(Pipeline.Workspace)/build/onnxruntime_gpu
|
|
artifact: whl
|
|
displayName: Republish artifacts |