|
| 1 | +#!/bin/bash |
| 2 | +set -e |
| 3 | + |
| 4 | +# FlagCX RPM package build script |
| 5 | +# Usage: ./build-flagcx-rpm.sh <backend> [base_image_version] |
| 6 | +# Supported backends: nvidia, metax, ascend |
| 7 | + |
| 8 | +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
| 9 | +PROJECT_DIR="$(dirname "$(dirname "$SCRIPT_DIR")")" |
| 10 | +BACKEND="${1:-}" |
| 11 | +BASE_IMAGE_VERSION="${2:-}" |
| 12 | + |
| 13 | +# Colors |
| 14 | +RED='\033[0;31m' |
| 15 | +GREEN='\033[0;32m' |
| 16 | +YELLOW='\033[1;33m' |
| 17 | +BLUE='\033[0;34m' |
| 18 | +NC='\033[0m' |
| 19 | + |
| 20 | +log_info() { echo -e "${GREEN}[INFO]${NC} $1"; } |
| 21 | +log_warn() { echo -e "${YELLOW}[WARN]${NC} $1"; } |
| 22 | +log_error() { echo -e "${RED}[ERROR]${NC} $1"; } |
| 23 | +log_step() { echo -e "${BLUE}[STEP]${NC} $1"; } |
| 24 | + |
| 25 | +# Show usage |
| 26 | +if [ -z "$BACKEND" ]; then |
| 27 | + log_error "No backend specified" |
| 28 | + echo "" |
| 29 | + echo "Usage: $0 <backend> [base_image_version]" |
| 30 | + echo "" |
| 31 | + echo "Supported backends:" |
| 32 | + echo " nvidia - Build RPM packages for NVIDIA GPUs" |
| 33 | + echo " metax - Build RPM packages for MetaX accelerators" |
| 34 | + echo " ascend - Build RPM packages for Ascend NPUs" |
| 35 | + echo "" |
| 36 | + echo "Examples:" |
| 37 | + echo " $0 nvidia" |
| 38 | + echo " $0 ascend 8.5.0-910-openeuler24.03-py3.11" |
| 39 | + exit 1 |
| 40 | +fi |
| 41 | + |
| 42 | +# Validate backend and set base image |
| 43 | +case "$BACKEND" in |
| 44 | + nvidia) |
| 45 | + BASE_IMAGE="nvcr.io/nvidia/cuda" |
| 46 | + [ -z "$BASE_IMAGE_VERSION" ] && BASE_IMAGE_VERSION="12.4.1-devel-rockylinux8" |
| 47 | + DOCKERFILE="${SCRIPT_DIR}/dockerfiles/Dockerfile.rpm.nvidia" |
| 48 | + ;; |
| 49 | + metax) |
| 50 | + BASE_IMAGE="rockylinux" |
| 51 | + [ -z "$BASE_IMAGE_VERSION" ] && BASE_IMAGE_VERSION="8" |
| 52 | + DOCKERFILE="${SCRIPT_DIR}/dockerfiles/Dockerfile.rpm.metax" |
| 53 | + ;; |
| 54 | + ascend) |
| 55 | + BASE_IMAGE="ascendai/cann" |
| 56 | + [ -z "$BASE_IMAGE_VERSION" ] && BASE_IMAGE_VERSION="8.5.0-910-openeuler24.03-py3.11" |
| 57 | + DOCKERFILE="${SCRIPT_DIR}/dockerfiles/Dockerfile.rpm.ascend" |
| 58 | + ;; |
| 59 | + *) |
| 60 | + log_error "Invalid backend: $BACKEND" |
| 61 | + echo "Supported backends: nvidia, metax, ascend" |
| 62 | + exit 1 |
| 63 | + ;; |
| 64 | +esac |
| 65 | + |
| 66 | +log_info "Building FlagCX RPM packages for $BACKEND backend" |
| 67 | +log_info "Using base image: ${BASE_IMAGE}:${BASE_IMAGE_VERSION}" |
| 68 | + |
| 69 | +# Sync changelog from CHANGELOG.md |
| 70 | +log_step "Synchronizing changelog..." |
| 71 | +if [ -f "${PROJECT_DIR}/packaging/sync-changelog.py" ]; then |
| 72 | + python3 "${PROJECT_DIR}/packaging/sync-changelog.py" || log_warn "Failed to sync changelog" |
| 73 | +else |
| 74 | + log_warn "sync-changelog.py not found, skipping changelog sync" |
| 75 | +fi |
| 76 | + |
| 77 | +# Build Docker image using backend-specific Dockerfile with shared RPM logic. |
| 78 | +log_step "Building Docker image..." |
| 79 | +docker build \ |
| 80 | + --network=host \ |
| 81 | + --build-arg BASE_IMAGE_VERSION="${BASE_IMAGE_VERSION}" \ |
| 82 | + -f "${DOCKERFILE}" \ |
| 83 | + -t "flagcx-rpm-${BACKEND}:${BASE_IMAGE_VERSION}" \ |
| 84 | + "${PROJECT_DIR}" |
| 85 | + |
| 86 | +# Extract RPM packages |
| 87 | +log_step "Extracting RPM packages..." |
| 88 | +OUTPUT_DIR="${PROJECT_DIR}/rpm-packages/${BACKEND}" |
| 89 | +mkdir -p "${OUTPUT_DIR}" |
| 90 | + |
| 91 | +CONTAINER_ID=$(docker create "flagcx-rpm-${BACKEND}:${BASE_IMAGE_VERSION}") |
| 92 | +docker cp "${CONTAINER_ID}:/root/rpmbuild/RPMS/" "${OUTPUT_DIR}/" |
| 93 | +docker cp "${CONTAINER_ID}:/root/rpmbuild/SRPMS/" "${OUTPUT_DIR}/" |
| 94 | +docker rm "${CONTAINER_ID}" |
| 95 | + |
| 96 | +# Fail loudly if no RPMs were extracted, so CI doesn't silently upload empty artifacts. |
| 97 | +if ! find "${OUTPUT_DIR}" -name '*.rpm' | grep -q .; then |
| 98 | + log_error "No RPM packages found under ${OUTPUT_DIR}" |
| 99 | + exit 1 |
| 100 | +fi |
| 101 | + |
| 102 | +log_info "✓ Packages built successfully for ${BACKEND}:" |
| 103 | +echo "" |
| 104 | +find "${OUTPUT_DIR}" -name "*.rpm" -exec ls -lh {} \; |
| 105 | + |
| 106 | +log_info "Build complete! Packages in: ${OUTPUT_DIR}" |
0 commit comments