mirror of
https://github.com/radixark/miles.git
synced 2026-11-01 07:53:37 +08:00
3547e67d3e
Co-authored-by: Jiajun Li <jiajun.li@radixark.ai> Co-authored-by: guapisolo <guapisolo@gmail.com>
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""Write release-lock.json for a release branch cut.
|
|
|
|
Usage: write_release_lock.py <branch> <sglang_sha> <megatron_sha> <ci_image_tag> <fp_cu130_x86> <fp_cu130_arm64> <fp_cu129_x86>
|
|
"""
|
|
|
|
import json
|
|
import sys
|
|
|
|
|
|
def main() -> int:
|
|
branch, sglang, megatron, image, fp_x86, fp_arm, fp_cu12 = sys.argv[1:8]
|
|
assert len(sglang) == 40 and len(megatron) == 40, "failed to resolve dependency SHAs"
|
|
json.dump(
|
|
{
|
|
"release_branch": branch,
|
|
"sglang_repo": "sgl-project/sglang",
|
|
"sglang_branch": "sglang-miles",
|
|
"sglang_commit": sglang,
|
|
"megatron_repo": "radixark/Megatron-LM",
|
|
"megatron_branch": "miles-main",
|
|
"megatron_commit": megatron,
|
|
"ci_image_tag": image,
|
|
# Audit only: wheels rolling tags can have assets replaced in place,
|
|
# so they cannot be pinned by name (see docker-build.yml).
|
|
"wheels_fingerprint_cu130_x86_64": fp_x86,
|
|
"wheels_fingerprint_cu130_aarch64": fp_arm,
|
|
"wheels_fingerprint_cu129_x86_64": fp_cu12,
|
|
},
|
|
open("release-lock.json", "w"),
|
|
indent=2,
|
|
)
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|