9f196058ae
* build: Minor tweeks for wheel build Signed-off-by: oliver könig <okoenig@nvidia.com> * ci: Workflows for wheel build 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> * build: Add CachedWheel Signed-off-by: oliver könig <okoenig@nvidia.com> * add version to init Signed-off-by: oliver könig <okoenig@nvidia.com> * revert Signed-off-by: oliver könig <okoenig@nvidia.com> * revert Signed-off-by: oliver könig <okoenig@nvidia.com> * revert Signed-off-by: oliver könig <okoenig@nvidia.com> * v2 Signed-off-by: oliver könig <okoenig@nvidia.com> * update Signed-off-by: oliver könig <okoenig@nvidia.com> * test Signed-off-by: oliver könig <okoenig@nvidia.com> * from packaging.version import parse Signed-off-by: oliver könig <okoenig@nvidia.com> * local version Signed-off-by: oliver könig <okoenig@nvidia.com> * remove file Signed-off-by: oliver könig <okoenig@nvidia.com> * revert Signed-off-by: oliver könig <okoenig@nvidia.com> * Updates and lint * revert missing cudaextension args Signed-off-by: oliver könig <okoenig@nvidia.com> * Add timeout * fix DG settings Signed-off-by: oliver könig <okoenig@nvidia.com> * DG_USE_LOCAL_VERSION Signed-off-by: oliver könig <okoenig@nvidia.com> * Update version * Detect local changes * Minor fix * Revert CUTLASS * Unify options --------- Signed-off-by: oliver könig <okoenig@nvidia.com> Co-authored-by: Chenggang Zhao <chenggangz@deepseek.com>
54 lines
1.5 KiB
YAML
54 lines
1.5 KiB
YAML
name: Build wheels
|
|
|
|
on:
|
|
workflow_dispatch:
|
|
inputs:
|
|
runs-on:
|
|
description: "The runner to use for the build"
|
|
required: true
|
|
type: string
|
|
default: ubuntu-22.04
|
|
python-version:
|
|
description: "The Python version to use for the build"
|
|
required: true
|
|
type: string
|
|
cuda-version:
|
|
description: "The CUDA version to use for the build"
|
|
required: true
|
|
type: string
|
|
torch-version:
|
|
description: "The PyTorch version to use for the build"
|
|
required: true
|
|
type: string
|
|
cxx11_abi:
|
|
description: "Enable torch flag C++11 ABI (TRUE/FALSE)"
|
|
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
|
|
use-local-version:
|
|
description: "Use local version"
|
|
required: false
|
|
type: boolean
|
|
default: false
|
|
|
|
jobs:
|
|
build-wheels:
|
|
uses: ./.github/workflows/_build.yml
|
|
with:
|
|
runs-on: ${{ inputs.runs-on }}
|
|
python-version: ${{ inputs.python-version }}
|
|
cuda-version: ${{ inputs.cuda-version }}
|
|
torch-version: ${{ inputs.torch-version }}
|
|
cxx11_abi: ${{ inputs.cxx11_abi }}
|
|
upload-to-release: ${{ inputs.upload-to-release }}
|
|
release-version: ${{ inputs.release-version }}
|
|
use-local-version: ${{ inputs.use-local-version }}
|