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

fix: typo

Browse Source
This commit is contained in:
Junya Morioka
2025-12-05 12:17:56 +09:00
parent 8fa80b072c
commit 3f85abe6ad
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-arm.yml
+2 -2
View File
@@ -8,7 +8,7 @@ jobs:
# Build wheels with self-hosted runner
# #########################################################
build_wheels_self_hosted:
name: Build wheels and Upload (Linux ARM64, self-hosted runner)
name: Build wheels and Upload (Linux ARM64)
uses: ./.github/workflows/_build_linux.yml
strategy:
fail-fast: false
@@ -18,7 +18,7 @@ jobs:
torch-version: ["2.9.1"]
# https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["13.0.1"]
runner: ["ubuntu-22.04-arm64"]
runner: ["ubuntu-22.04-arm"]
with:
flash-attn-version: ${{ matrix.flash-attn-version }}
python-version: ${{ matrix.python-version }}