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

Update common.py

Browse Source 
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Junya Morioka
2025-11-02 02:45:35 +09:00
committed by GitHub
parent 1696c1f516
commit 8826f91599
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
common.py
+7
View File
@@ -3,6 +3,13 @@ import re
def parse_wheel_filename(filename: str) -> dict | None:
"""
Extract information from a wheel filename.
Examples:
flash_attn-2.6.3+cu124torch2.5-cp311-cp311-linux_x86_64.whl
flash_attn-2.7.4+cu124torch2.6-cp311-cp311-linux_x86_64.whl
flash_attn-2.7.4.post1+cu130torch2.9-cp310-cp310-linux_x86_64.whl
---
Wheel filename から情報を抽出
例: flash_attn-2.6.3+cu124torch2.5-cp311-cp311-linux_x86_64.whl
flash_attn-2.7.4+cu124torch2.6-cp311-cp311-linux_x86_64.whl