mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:27:54 -04:00
chore: update section design
This commit is contained in:
@@ -48,7 +48,7 @@ pip install ./flash_attn-2.6.3+cu124torch2.5-cp312-cp312-linux_x86_64.whl
|
||||
|
||||
## Packages
|
||||
|
||||
### Linux x86_64
|
||||
### 🐧 Linux x86_64
|
||||
|
||||
#### Flash-Attention 2.8.3
|
||||
|
||||
@@ -935,7 +935,7 @@ pip install ./flash_attn-2.6.3+cu124torch2.5-cp312-cp312-linux_x86_64.whl
|
||||
|
||||
</details>
|
||||
|
||||
### Windows x86_64
|
||||
### 🪟 Windows x86_64
|
||||
|
||||
#### Flash-Attention 2.8.3
|
||||
|
||||
|
||||
@@ -319,6 +319,17 @@ def merge_duplicate_rows(df: pd.DataFrame) -> pd.DataFrame:
|
||||
return merged_df
|
||||
|
||||
|
||||
def get_os_emoji(os_name: str) -> str:
|
||||
"""Get emoji for OS name."""
|
||||
os_lower = os_name.lower()
|
||||
if "linux" in os_lower:
|
||||
return "🐧 "
|
||||
elif "windows" in os_lower:
|
||||
return "🪟 "
|
||||
else:
|
||||
return ""
|
||||
|
||||
|
||||
def generate_markdown_table_by_os(df: pd.DataFrame) -> str:
|
||||
"""Generate markdown tables grouped by OS and Flash-Attention version."""
|
||||
if df.empty:
|
||||
@@ -339,8 +350,9 @@ def generate_markdown_table_by_os(df: pd.DataFrame) -> str:
|
||||
cuda_ascending=True,
|
||||
)
|
||||
|
||||
# Create OS section header
|
||||
os_lines = [f"### {os_name}", ""]
|
||||
# Create OS section header with emoji
|
||||
os_emoji = get_os_emoji(os_name)
|
||||
os_lines = [f"### {os_emoji}{os_name}", ""]
|
||||
|
||||
# Group by Flash-Attention version within each OS
|
||||
fa_versions = []
|
||||
|
||||
Reference in New Issue
Block a user