[MUSA][11/N] ci: add MUSA 4.3 kernel build and release pipeline (#18537)

Co-authored-by: ximin.chen <ximin.chen@mthreads.com>
This commit is contained in:
johnnycxm
2026-02-26 07:45:12 -08:00
committed by GitHub
co-authored by ximin.chen
parent e55e65535e
commit 5939b8912a
4 changed files with 169 additions and 11 deletions
+5
View File
@@ -0,0 +1,5 @@
#!/bin/bash
set -euo pipefail
PIP_INSTALL="python3 -m pip install --no-cache-dir"
${PIP_INSTALL} --upgrade pip setuptools torchada
+46
View File
@@ -0,0 +1,46 @@
#!/usr/bin/env bash
set -euo pipefail
# Rename MUSA wheels to include a +musa<suffix> build tag.
# Usage:
# rename_wheels_musa.sh <musa_suffix> [wheel_dir]
# Example:
# rename_wheels_musa.sh 43 sgl-kernel/dist
if [[ $# -lt 1 || $# -gt 2 ]]; then
echo "Usage: $0 <musa_suffix> [wheel_dir]" >&2
exit 1
fi
MUSA_SUFFIX="$1"
WHEEL_DIR="${2:-dist}"
wheel_files=("$WHEEL_DIR"/*.whl)
if [[ ! -e "${wheel_files[0]}" ]]; then
echo "No wheel files found in ${WHEEL_DIR}/, nothing to rename."
exit 0
fi
for wheel in "${wheel_files[@]}"; do
# Normalize platform tag to manylinux2014
intermediate_wheel="${wheel/linux/manylinux2014}"
# Extract Python ABI version (e.g. cp310)
if [[ $intermediate_wheel =~ -cp([0-9]+)- ]]; then
cp_version="${BASH_REMATCH[1]}"
else
echo "Could not extract Python version from wheel name: $intermediate_wheel" >&2
continue
fi
# Insert +musa<suffix> before the Python ABI tag
new_wheel="${intermediate_wheel/-cp${cp_version}/+musa${MUSA_SUFFIX}-cp${cp_version}}"
if [[ "$wheel" != "$new_wheel" ]]; then
echo "Renaming $wheel -> $new_wheel"
mv -- "$wheel" "$new_wheel"
fi
done
echo "MUSA wheel renaming completed."