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

chore: add environment test

Browse Source
This commit is contained in:
Junya Morioka
2025-12-05 12:05:13 +09:00
parent 29a4910983
commit 9e2a3c1ed4
1 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-utils.yml
+2 -8
View File
@@ -10,13 +10,7 @@ jobs:
fail-fast: false
matrix:
python-version: ["3.13", "3.12"]
runner:
[
"ubuntu-22.04-arm",
"ubuntu-22.04",
"self-hosted",
"openci-runner-beta",
]
runner: ["ubuntu-22.04-arm", "ubuntu-22.04", "self-hosted"]
runs-on: ${{ matrix.runner }}
steps:
- name: Check environment
@@ -48,7 +42,7 @@ jobs:
# - MAX_JOBS x NVCC_THREADS(<= 4) <= NUM_THREADS
# - 2.5GB x MAX_JOBS x NVCC_THREADS(<= 4) <= RAM_GB
MAX_PRODUCT_CPU=$NUM_THREADS
MAX_PRODUCT_RAM=$(awk -v ram="$RAM_GB" 'BEGIN {print int(ram / 2.5)}')
MAX_PRODUCT_RAM=$(awk -v ram="$RAM_GB" 'BEGIN {print int(ram / 3.0)}')
MAX_PRODUCT=$((MAX_PRODUCT_CPU < MAX_PRODUCT_RAM ? MAX_PRODUCT_CPU : MAX_PRODUCT_RAM))
# Set MAX_JOBS and NVCC_THREADS so that MAX_JOBS x NVCC_THREADS ≈ MAX_PRODUCT with NVCC_THREADS <= 4