mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:27:54 -04:00
Junya Morioka
86938251af
- Move reusable functions (parse_numeric_version, normalize_semantic_version, get_tag_from_url, get_os_emoji, collect_versions_from_assets, format_versions) to common.py for code reuse across scripts - Add comprehensive docstrings to all modules and functions for better maintainability - Enhance parse_wheel_filename to support multiple manylinux tags (e.g., manylinux_2_24_x86_64.manylinux_2_28_x86_64) - Improve normalize_platform_name to handle manylinux formats with multiple tags - Refactor create_packages.py to use shared utilities from common.py - Refactor create_release_history.py with improved function naming and error handling - Refactor create_release_note.py with better error messages and usage documentation - Decode URL encoding for improved readability in package documentation - Update docs/packages.md with formatted output from refactored scripts
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
"""Generate release notes from assets.json.
|
|
|
|
This script generates markdown release notes from a GitHub release assets JSON file.
|
|
It extracts version information from wheel filenames and creates a formatted table
|
|
showing supported Flash-Attention, Python, PyTorch, and CUDA versions for each platform.
|
|
|
|
Usage:
|
|
python create_release_note.py <assets.json>
|
|
|
|
Arguments:
|
|
assets.json: Path to JSON file containing GitHub release assets
|
|
(obtained via `gh release view --json assets`)
|
|
|
|
Output:
|
|
Markdown-formatted release notes to stdout
|
|
|
|
Example:
|
|
gh release view v0.7.0 --json assets > assets.json
|
|
python create_release_note.py assets.json > release_notes.md
|
|
"""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from common import collect_versions_from_assets, format_versions, load_assets_json
|
|
|
|
|
|
def generate_release_notes(assets: list[dict]) -> str:
|
|
"""Generate release notes from assets.
|
|
|
|
Args:
|
|
assets: List of asset dictionaries
|
|
|
|
Returns:
|
|
Formatted release notes as markdown string
|
|
"""
|
|
versions_by_platform = collect_versions_from_assets(assets)
|
|
|
|
if not versions_by_platform:
|
|
return ""
|
|
|
|
notes = []
|
|
for platform_name in sorted(versions_by_platform.keys()):
|
|
data = versions_by_platform[platform_name]
|
|
|
|
notes.append(f"## {platform_name}")
|
|
notes.append("")
|
|
notes.append("| Flash-Attention | Python | PyTorch | CUDA |")
|
|
notes.append("| --- | --- | --- | --- |")
|
|
|
|
flash_versions = format_versions(data["flash_versions"])
|
|
python_versions = format_versions(data["python_versions"])
|
|
torch_versions = format_versions(data["torch_versions"])
|
|
cuda_versions = format_versions(data["cuda_versions"])
|
|
|
|
notes.append(
|
|
f"| {flash_versions} | {python_versions} | {torch_versions} | {cuda_versions} |"
|
|
)
|
|
notes.append("")
|
|
|
|
return "\n".join(notes)
|
|
|
|
|
|
def main():
|
|
try:
|
|
if len(sys.argv) != 2:
|
|
print("Usage: python create_release_note.py <assets.json>", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
assets_json_path = Path(sys.argv[1])
|
|
if not assets_json_path.exists():
|
|
print(f"File not found: {assets_json_path}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
assets = load_assets_json(assets_json_path)
|
|
if not assets:
|
|
print("No assets found in JSON file", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
text = generate_release_notes(assets)
|
|
if text:
|
|
print(text)
|
|
else:
|
|
print("No wheel assets found", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|