Skip to content

Commit

Permalink
Adding a new pipeline for publishing to Python Cuda 12 packages. (#18712
Browse files Browse the repository at this point in the history
)

### 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. -->
  • Loading branch information
jchen351 authored Dec 11, 2023
1 parent bfa5eb4 commit ce1fed6
Show file tree
Hide file tree
Showing 2 changed files with 75 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -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
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit ce1fed6

Please sign in to comment.