mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-28 20:11:22 +00:00
Adding a new pipeline for publishing to Python Cuda 12 packages. (#18712)
### 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
bfa5eb4591
commit
ce1fed6ddf
2 changed files with 75 additions and 0 deletions
|
|
@ -0,0 +1,24 @@
|
|||
parameters:
|
||||
- name: nightly
|
||||
type: string
|
||||
default: '1'
|
||||
- name: build_id
|
||||
type: string
|
||||
default: 'latest'
|
||||
- name: project
|
||||
type: string
|
||||
default: 'Lotus'
|
||||
- name: pipeline
|
||||
type: string
|
||||
default: 'Python-CUDA-Packaging-Pipeline'
|
||||
|
||||
stages:
|
||||
- template: stages/py-cuda-publishing-stage.yml
|
||||
parameters:
|
||||
build_id: ${{ parameters.build_id }}
|
||||
project: ${{ parameters.project }}
|
||||
pipeline: ${{ parameters.pipeline }}
|
||||
${{ if ne(parameters.nightly, '1') }}:
|
||||
artifact_feed: onnxruntime-cuda-12
|
||||
${{ else }}:
|
||||
artifact_feed: ort-cuda-12-nightly
|
||||
|
|
@ -0,0 +1,51 @@
|
|||
parameters:
|
||||
- name: build_id
|
||||
type: string
|
||||
- name: project
|
||||
type: string
|
||||
- name: pipeline
|
||||
type: string
|
||||
- name: artifact_feed
|
||||
type: string
|
||||
default: 'onnxruntime-cuda-12'
|
||||
- name: dependencies
|
||||
type: string
|
||||
default: 'none'
|
||||
|
||||
stages:
|
||||
- stage: Python_Publishing
|
||||
${{ if ne(parameters.dependencies, 'none') }}:
|
||||
dependsOn: ${{ parameters.dependencies }}
|
||||
${{ if eq(parameters.dependencies, 'none') }}:
|
||||
dependsOn: []
|
||||
jobs:
|
||||
- job:
|
||||
pool: 'onnxruntime-Ubuntu2004-AMD-CPU'
|
||||
steps:
|
||||
- checkout: none
|
||||
- task: DownloadPipelineArtifact@2
|
||||
inputs:
|
||||
artifact: 'onnxruntime_gpu'
|
||||
targetPath: '$(Build.SourcesDirectory)/onnxruntime-gpu'
|
||||
${{ if ne(parameters.build_id, 'latest') }}:
|
||||
buildType: 'specific'
|
||||
project: '${{ parameters.project }}'
|
||||
pipeline: '${{ parameters.pipeline }}'
|
||||
buildVersionToDownload: 'specific'
|
||||
buildId: '${{ parameters.build_id }}'
|
||||
displayName: 'Download Build Artifacts - onnxruntime-gpu'
|
||||
- task: UsePythonVersion@0
|
||||
displayName: 'Use Python 3.x'
|
||||
- script: 'pip install twine==3.4.2'
|
||||
displayName: 'Install Twine'
|
||||
- task: TwineAuthenticate@1
|
||||
displayName: 'Twine Authenticate '
|
||||
inputs:
|
||||
artifactFeed: PublicPackages/${{ parameters.artifact_feed }}
|
||||
- script: 'python -m twine upload -r ${{ parameters.artifact_feed }} --config-file $(PYPIRC_PATH) --non-interactive --skip-existing *.whl'
|
||||
workingDirectory: '$(Build.SourcesDirectory)/onnxruntime-gpu'
|
||||
displayName: 'Uploading wheels to ${{ parameters.artifact_feed }}'
|
||||
retryCountOnTaskFailure: 3
|
||||
env:
|
||||
SYSTEM_ACCESSTOKEN: $(System.AccessToken)
|
||||
|
||||
Loading…
Reference in a new issue