yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

[WIP] workflow_call

Browse Source
This commit is contained in:
Junya Morioka
2025-08-10 15:58:16 +09:00
parent 72396e7ce3
commit 03dfdc4389
4 changed files with 17 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build_linux.yml
-3
View File
@@ -23,9 +23,6 @@ on:
description: "CUDA version"
required: true
type: string
secrets:
GITHUB_TOKEN:
required: true
jobs:
build_wheels:
.github/workflows/build_linux_self_host.yml
-3
View File
@@ -23,9 +23,6 @@ on:
description: "CUDA version"
required: true
type: string
secrets:
GITHUB_TOKEN:
required: true
jobs:
build_wheels_self_hosted:
.github/workflows/build_test.yml
+17 -22
View File
@@ -7,9 +7,8 @@ jobs:
# #########################################################
# Build wheels with GitHub hosted runner
# #########################################################
build_wheels:
name: Build Root
runs-on: ubuntu-latest
build_wheels_linux:
name: Build Linux
strategy:
fail-fast: false
matrix:
@@ -18,18 +17,16 @@ jobs:
torch-version: ["2.5.1"]
# https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["12.4.1"]
steps:
- uses: actions/checkout@v4
uses: ./.github/workflows/build_linux.yml
with:
flash-attn-version: ${{ matrix.flash-attn-version }}
python-version: ${{ matrix.python-version }}
torch-version: ${{ matrix.torch-version }}
cuda-version: ${{ matrix.cuda-version }}
secrets: inherit
- uses: ./.github/workflows/build_linux.yml
with:
flash-attn-version: ${{ matrix.flash-attn-version }}
python-version: ${{ matrix.python-version }}
torch-version: ${{ matrix.torch-version }}
cuda-version: ${{ matrix.cuda-version }}
build_wheels_windows:
name: Build Root
runs-on: ubuntu-latest
name: Build Windows
strategy:
fail-fast: false
matrix:
@@ -38,12 +35,10 @@ jobs:
torch-version: ["2.5.1"]
# https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["12.4.1"]
steps:
- uses: actions/checkout@v4
- uses: ./.github/workflows/build_windows.yml
with:
flash-attn-version: ${{ matrix.flash-attn-version }}
python-version: ${{ matrix.python-version }}
torch-version: ${{ matrix.torch-version }}
cuda-version: ${{ matrix.cuda-version }}
uses: ./.github/workflows/build_windows.yml
with:
flash-attn-version: ${{ matrix.flash-attn-version }}
python-version: ${{ matrix.python-version }}
torch-version: ${{ matrix.torch-version }}
cuda-version: ${{ matrix.cuda-version }}
secrets: inherit
.github/workflows/build_windows.yml
-3
View File
@@ -23,9 +23,6 @@ on:
description: "CUDA version"
required: true
type: string
secrets:
GITHUB_TOKEN:
required: true
jobs:
build_windows_wheels: