93b3c28fa8
* build: Allow NGC builds Signed-off-by: oliver könig <okoenig@nvidia.com> * reduce grid Signed-off-by: oliver könig <okoenig@nvidia.com> * update grid Signed-off-by: oliver könig <okoenig@nvidia.com> * fix Signed-off-by: oliver könig <okoenig@nvidia.com> * upgrade cuda action Signed-off-by: oliver könig <okoenig@nvidia.com> * remove test Signed-off-by: oliver könig <okoenig@nvidia.com> * py3.8 Signed-off-by: oliver könig <okoenig@nvidia.com> * fix Signed-off-by: oliver könig <okoenig@nvidia.com> * exclude Signed-off-by: oliver könig <okoenig@nvidia.com> * fix Signed-off-by: oliver könig <okoenig@nvidia.com> * torch-version Signed-off-by: oliver könig <okoenig@nvidia.com> * py3.8/torch2.1/cuda12.3 Signed-off-by: oliver könig <okoenig@nvidia.com> * Update publish.yml * fix grid Signed-off-by: oliver könig <okoenig@nvidia.com> * fix Signed-off-by: oliver könig <okoenig@nvidia.com> * cuda11.8 Signed-off-by: oliver könig <okoenig@nvidia.com> * no hopper for 118 Signed-off-by: oliver könig <okoenig@nvidia.com> * fix Signed-off-by: oliver könig <okoenig@nvidia.com> * fix Signed-off-by: oliver könig <okoenig@nvidia.com> --------- Signed-off-by: oliver könig <okoenig@nvidia.com>
35 lines
961 B
YAML
35 lines
961 B
YAML
name: Build wheels in a container
|
|
|
|
on:
|
|
workflow_dispatch:
|
|
inputs:
|
|
runs-on:
|
|
description: "The runner to use for the build"
|
|
required: true
|
|
type: string
|
|
default: ubuntu-22.04
|
|
container-image:
|
|
description: "Container image"
|
|
required: true
|
|
type: string
|
|
upload-to-release:
|
|
description: "Upload wheel to this release"
|
|
required: false
|
|
type: boolean
|
|
default: false
|
|
release-version:
|
|
description: "Upload wheel to this release"
|
|
required: false
|
|
type: string
|
|
|
|
push:
|
|
|
|
jobs:
|
|
build-wheels:
|
|
uses: ./.github/workflows/_build_in_container.yml
|
|
with:
|
|
runs-on: ${{ inputs.runs-on || 'ubuntu-22.04' }}
|
|
container-image: ${{ inputs.container-image || 'nvcr.io/nvidia/pytorch:25.06-py3' }}
|
|
upload-to-release: ${{ inputs.upload-to-release || false }}
|
|
release-version: ${{ inputs.release-version || 'v2.2.5' }}
|