This repository has been archived by the owner on Aug 10, 2024. It is now read-only.

Build Triton wheel #1

	name: Build Triton wheel

	jobs:
	build:
	runs-on: ubuntu-latest

	steps:
	- name: Checkout
	uses: actions/checkout@v4
	with:
	ref: ${{ github.event.inputs.ref }}
	repository: ${{ github.event.inputs.repository }}

	- name: Apply patches
	run: \|
	set -e

	# List of patches
	patches=(
	"https://raw.githubusercontent.com/${{ github.repository }}/${{ github.sha }}/patches/triton/0000-add-support-for-conversion-fp16-to-fp32.patch"
	"https://raw.githubusercontent.com/${{ github.repository }}/${{ github.sha }}/patches/triton/0000-fix-max-ptx-version.patch"
	)

	# Apply patches
	for patch in $patches; do
	curl "$patch" \| patch -p1
	done

	- name: Build wheels
	uses: pypa/[email protected]
	env:
	CIBW_BUILD: cp311-manylinux_x86_64
	CIBW_ENVIRONMENT: "TRITON_BUILD_WITH_CLANG_LLD=1"

	- name: Create release
	uses: softprops/action-gh-release@v2
	with:
	files: wheelhouse/*.whl
	tag_name: ${{ github.event.inputs.tag_name }}

	on:
	workflow_dispatch:
	inputs:
	repository:
	default: triton-lang/triton
	description: Source repository
	required: true
	type: string

	ref:
	default: release/2.3.x
	description: Source ref
	required: true
	type: string

	tag_name:
	description: Target tag
	required: true
	type: string

	permissions:
	contents: write

Provide feedback