Skip to content

Commit

Permalink
(triton) Bring back to CI
Browse files Browse the repository at this point in the history
  • Loading branch information
sasha0552 authored Oct 7, 2024
1 parent dca6765 commit ccf8c2f
Show file tree
Hide file tree
Showing 18 changed files with 526 additions and 6 deletions.
44 changes: 44 additions & 0 deletions .github/actions/build-triton/action.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
name: Build Triton wheels

inputs:
python_version:
required: true
type: string

ref:
default: main
type: string

repository:
default: triton-lang/triton
type: string

runs:
using: "composite"

steps:
- name: Checkout ${{ inputs.repository }} (${{ inputs.ref }})
uses: actions/checkout@v4
with:
ref: ${{ inputs.ref }}
repository: ${{ inputs.repository }}
path: ${{ inputs.repository }}/${{ inputs.ref }}

- name: Apply patches
uses: ./.github/actions/common/apply-patches
with:
ref: ${{ inputs.ref }}
repository: ${{ inputs.repository }}

- name: Resolve SOURCE_DATE_EPOCH
shell: bash
run: echo "SOURCE_DATE_EPOCH=$(git log -1 --format=%ct)" | tee "$GITHUB_ENV"

- name: Build wheels
uses: pypa/cibuildwheel@v2.21.2
env:
CIBW_BUILD: cp${{ inputs.python_version }}-manylinux_x86_64
CIBW_BUILD_VERBOSITY: 1
with:
package-dir: ${{ inputs.repository }}/${{ inputs.ref }}/python
output-dir: dist
59 changes: 59 additions & 0 deletions .github/workflows/triton-pascal-2.2.0.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
name: "[X] Build Triton (v2.2.0)"

jobs:
build:
runs-on: ubuntu-latest

env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Build vLLM wheels
uses: ./.github/actions/build-triton
with:
python_version: ${{ matrix.python_version }}
ref: 0e7b97bd47fc4beb21ae960a516cd9a7ae9bc060

- name: Repackage wheels
uses: ./.github/actions/common/repackage-wheels
with:
name: triton_pascal
path: dist/*.whl

- name: Upload artifact
uses: actions/upload-artifact@v4
with:
name: triton_pascal-${{ matrix.python_version }}
path: dist/*.whl

strategy:
matrix:
python_version:
- "37"
- "38"
- "39"
- "310"
- "311"
- "312"

publish:
needs: build
runs-on: ubuntu-latest

permissions:
contents: write

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Publish wheels
uses: ./.github/actions/common/publish-wheels
with:
prefix: triton_pascal-2.2.0-

on:
workflow_dispatch:
59 changes: 59 additions & 0 deletions .github/workflows/triton-pascal-2.3.0.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
name: "[X] Build Triton (v2.2.0)"

jobs:
build:
runs-on: ubuntu-latest

env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Build vLLM wheels
uses: ./.github/actions/build-triton
with:
python_version: ${{ matrix.python_version }}
ref: 3f8d91bb17f6e7bc33dc995ae0860db89d351c7b

- name: Repackage wheels
uses: ./.github/actions/common/repackage-wheels
with:
name: triton_pascal
path: dist/*.whl

- name: Upload artifact
uses: actions/upload-artifact@v4
with:
name: triton_pascal-${{ matrix.python_version }}
path: dist/*.whl

strategy:
matrix:
python_version:
- "37"
- "38"
- "39"
- "310"
- "311"
- "312"

publish:
needs: build
runs-on: ubuntu-latest

permissions:
contents: write

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Publish wheels
uses: ./.github/actions/common/publish-wheels
with:
prefix: triton_pascal-2.3.0-

on:
workflow_dispatch:
59 changes: 59 additions & 0 deletions .github/workflows/triton-pascal-2.3.1.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,59 @@
name: "[X] Build Triton (v2.2.0)"

jobs:
build:
runs-on: ubuntu-latest

env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Build vLLM wheels
uses: ./.github/actions/build-triton
with:
python_version: ${{ matrix.python_version }}
ref: 958fccea74da58e7e0595ab88ae6cd3f6795a173

- name: Repackage wheels
uses: ./.github/actions/common/repackage-wheels
with:
name: triton_pascal
path: dist/*.whl

- name: Upload artifact
uses: actions/upload-artifact@v4
with:
name: triton_pascal-${{ matrix.python_version }}
path: dist/*.whl

strategy:
matrix:
python_version:
- "37"
- "38"
- "39"
- "310"
- "311"
- "312"

publish:
needs: build
runs-on: ubuntu-latest

permissions:
contents: write

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Publish wheels
uses: ./.github/actions/common/publish-wheels
with:
prefix: triton_pascal-2.3.1-

on:
workflow_dispatch:
58 changes: 58 additions & 0 deletions .github/workflows/triton-pascal-3.0.0.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
name: "[X] Build Triton (v3.0.0)"

jobs:
build:
runs-on: ubuntu-latest

env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Build vLLM wheels
uses: ./.github/actions/build-triton
with:
python_version: ${{ matrix.python_version }}
ref: 55a4ab051c88ba2baa031e520a339d3fded6468f

- name: Repackage wheels
uses: ./.github/actions/common/repackage-wheels
with:
name: triton_pascal
path: dist/*.whl

- name: Upload artifact
uses: actions/upload-artifact@v4
with:
name: triton_pascal-${{ matrix.python_version }}
path: dist/*.whl

strategy:
matrix:
python_version:
- "38"
- "39"
- "310"
- "311"
- "312"

publish:
needs: build
runs-on: ubuntu-latest

permissions:
contents: write

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Publish wheels
uses: ./.github/actions/common/publish-wheels
with:
prefix: triton_pascal-3.0.0-

on:
workflow_dispatch:
49 changes: 49 additions & 0 deletions .github/workflows/vllm-pascal-0.5.5.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
name: "[X] Build vLLM (v0.5.5)"

jobs:
build:
runs-on: ubuntu-latest

env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Build vLLM wheels
uses: ./.github/actions/build-vllm
with:
ref: v0.5.5

- name: Repackage wheels
uses: ./.github/actions/common/repackage-wheels
with:
name: vllm_pascal
tags: cp38-abi3-linux_x86_64
path: dist/*.whl

- name: Upload artifact
uses: actions/upload-artifact@v4
with:
name: vllm_pascal
path: dist/*.whl

publish:
needs: build
runs-on: ubuntu-latest

permissions:
contents: write

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Publish wheels
uses: ./.github/actions/common/publish-wheels
with:
prefix: vllm_pascal-0.6.2-

on:
workflow_dispatch:
49 changes: 49 additions & 0 deletions .github/workflows/vllm-pascal-0.6.0.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
name: "[X] Build vLLM (v0.6.0)"

jobs:
build:
runs-on: ubuntu-latest

env:
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Build vLLM wheels
uses: ./.github/actions/build-vllm
with:
ref: v0.6.0

- name: Repackage wheels
uses: ./.github/actions/common/repackage-wheels
with:
name: vllm_pascal
tags: cp38-abi3-linux_x86_64
path: dist/*.whl

- name: Upload artifact
uses: actions/upload-artifact@v4
with:
name: vllm_pascal
path: dist/*.whl

publish:
needs: build
runs-on: ubuntu-latest

permissions:
contents: write

steps:
- name: Checkout
uses: actions/checkout@v4

- name: Publish wheels
uses: ./.github/actions/common/publish-wheels
with:
prefix: vllm_pascal-0.6.2-

on:
workflow_dispatch:
Loading

0 comments on commit ccf8c2f

Please sign in to comment.