Embed DeepGEMM source (not submodule) for SM100 raw CUDA GEMM primitives
This commit is contained in:
53
third_party/DeepGEMM/.github/workflows/build.yml
vendored
Normal file
53
third_party/DeepGEMM/.github/workflows/build.yml
vendored
Normal file
@@ -0,0 +1,53 @@
|
||||
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 }}
|
||||
Reference in New Issue
Block a user