Update package. Pack package as nix package

This commit is contained in:
2026-04-11 13:01:37 +00:00
parent 8d8db0e972
commit ca6342eca6
7 changed files with 286 additions and 0 deletions

162
pack/pack_ascend_toolkit.sh Executable file
View File

@@ -0,0 +1,162 @@
#!/usr/bin/env bash
#
# pack_ascend_toolkit.sh — создаёт два tar.gz архива из собранного ascend-toolkit:
#
# 1. ascend-toolkit-8.5.0-full.tar.gz — полный архив (все файлы)
# 2. ascend-toolkit-8.5.0-minimal.tar.gz — минимальный набор для режима sim
#
# Использование:
# ./pack_ascend_toolkit.sh [path_to_result_dir]
#
# Если аргумент не указан, используется ./result
set -euo pipefail
RESULT_DIR="${1:-$(cd "$(dirname "$0")" && ls -d result 2>/dev/null || echo result)}"
VERSION="8.5.0"
OUTPUT_DIR="${2:-$(dirname "$0")}"
if [ ! -d "$RESULT_DIR" ]; then
echo "[ERROR] Directory not found: $RESULT_DIR"
exit 1
fi
cd "$RESULT_DIR"
# =============================================================================
# 1. Полный архив — всё содержимое
# =============================================================================
echo "=== Creating FULL archive ==="
FULL_ARCHIVE="${OUTPUT_DIR}/ascend-toolkit-${VERSION}-full.tar.gz"
tar czf "$FULL_ARCHIVE" .
FULL_SIZE=$(wc -c < "$FULL_ARCHIVE" | numfmt --to=iec-i --suffix=B)
echo " FULL archive: $FULL_ARCHIVE ($FULL_SIZE)"
# =============================================================================
# 2. Минимальный архив — только файлы для режима sim
# =============================================================================
echo "=== Creating MINIMAL archive (sim mode) ==="
# Создаём временную директорию с минимальным набором
MINIMAL_DIR=$(mktemp -d)
trap "rm -rf $MINIMAL_DIR" EXIT
# Копируем только нужные для sim-режима файлы/директории
#
# Структура основана на том, что реально используется при
# bash run.sh -r sim -v Ascend310P3
#
# Основные потребители:
# - cmake/npu_lib.cmake → compiler/tikcpp/ascendc_kernel_cmake
# - bisheng компиляция → tools/bisheng_compiler
# - cce линковка → tools/ccec_compiler (через x86_64-linux/ccec_compiler symlink)
# - симулятор → tools/simulator/Ascend310P3/lib
# - заголовки → include
# - библиотеки → lib64, devlib
# - OPP данные → opp/built-in (нужны для op_impl)
# - Python bindings → python/site-packages
# - Runtime → runtime
# - Toolkit → toolkit
# - Конфигурации → conf, compat
# - set_env.sh — корень
# =============================================================================
echo " Copying essential files..."
# Корневые файлы
mkdir -p "$MINIMAL_DIR"
cp -a set_env.sh "$MINIMAL_DIR/" 2>/dev/null || true
# bin — содержит setenv.bash symlink + ключевые утилиты
mkdir -p "$MINIMAL_DIR/bin"
cp -a bin/* "$MINIMAL_DIR/bin/" 2>/dev/null || true
# compiler — tikcpp cmake модули
mkdir -p "$MINIMAL_DIR/compiler"
cp -a compiler/tikcpp "$MINIMAL_DIR/compiler/" 2>/dev/null || true
# compat
mkdir -p "$MINIMAL_DIR/compat"
cp -a compat/* "$MINIMAL_DIR/compat/" 2>/dev/null || true
# conf
mkdir -p "$MINIMAL_DIR/conf"
cp -a conf/* "$MINIMAL_DIR/conf/" 2>/dev/null || true
# devlib — stub библиотеки для разработки
mkdir -p "$MINIMAL_DIR/devlib"
cp -a devlib/* "$MINIMAL_DIR/devlib/" 2>/dev/null || true
# fwkacllib — framework ACL библиотеки
mkdir -p "$MINIMAL_DIR/fwkacllib"
cp -a fwkacllib/* "$MINIMAL_DIR/fwkacllib/" 2>/dev/null || true
# include — все заголовочные файлы (критично для компиляции)
mkdir -p "$MINIMAL_DIR/include"
cp -a include/* "$MINIMAL_DIR/include/" 2>/dev/null || true
# lib64 — основные библиотеки
mkdir -p "$MINIMAL_DIR/lib64"
cp -a lib64/* "$MINIMAL_DIR/lib64/" 2>/dev/null || true
# pkg_inc
mkdir -p "$MINIMAL_DIR/pkg_inc"
cp -a pkg_inc/* "$MINIMAL_DIR/pkg_inc/" 2>/dev/null || true
# python
mkdir -p "$MINIMAL_DIR/python"
cp -a python/* "$MINIMAL_DIR/python/" 2>/dev/null || true
# runtime
mkdir -p "$MINIMAL_DIR/runtime"
cp -a runtime/* "$MINIMAL_DIR/runtime/" 2>/dev/null || true
# toolkit
mkdir -p "$MINIMAL_DIR/toolkit"
cp -a toolkit/* "$MINIMAL_DIR/toolkit/" 2>/dev/null || true
# x86_64-linux — содержит архитектурно-зависимые файлы (symlinks на реальные директории)
mkdir -p "$MINIMAL_DIR/x86_64-linux"
cp -a x86_64-linux/* "$MINIMAL_DIR/x86_64-linux/" 2>/dev/null || true
# tools — выборочно:
# - bisheng_compiler — основной AscendC компилятор
# - ccec_compiler — linker для AI Core
# - simulator — библиотека симулятора для sim режима
# - tikcpp — tikcpp файлы
# - aml — используется в LD_LIBRARY_PATH
# - profiler — может использоваться
# - ascendc_tools — ascendc инструменты
mkdir -p "$MINIMAL_DIR/tools"
cp -a tools/bisheng_compiler "$MINIMAL_DIR/tools/" 2>/dev/null || true
cp -a tools/ccec_compiler "$MINIMAL_DIR/tools/" 2>/dev/null || true
cp -a tools/simulator "$MINIMAL_DIR/tools/" 2>/dev/null || true
cp -a tools/tikcpp "$MINIMAL_DIR/tools/" 2>/dev/null || true
cp -a tools/aml "$MINIMAL_DIR/tools/" 2>/dev/null || true
cp -a tools/profiler "$MINIMAL_DIR/tools/" 2>/dev/null || true
cp -a tools/ascendc_tools "$MINIMAL_DIR/tools/" 2>/dev/null || true
# opp — operator packages
# Для sim режима нужны op_impl (tiling и kernel реализации)
mkdir -p "$MINIMAL_DIR/opp"
cp -a opp/built-in "$MINIMAL_DIR/opp/" 2>/dev/null || true
cp -a opp/include "$MINIMAL_DIR/opp/" 2>/dev/null || true
cp -a opp/lib64 "$MINIMAL_DIR/opp/" 2>/dev/null || true
echo " Compressing..."
MINIMAL_ARCHIVE="${OUTPUT_DIR}/ascend-toolkit-${VERSION}-minimal.tar.gz"
tar czf "$MINIMAL_ARCHIVE" -C "$MINIMAL_DIR" .
MINIMAL_SIZE=$(wc -c < "$MINIMAL_ARCHIVE" | numfmt --to=iec-i --suffix=B)
echo " MINIMAL archive: $MINIMAL_ARCHIVE ($MINIMAL_SIZE)"
# =============================================================================
# Итог
# =============================================================================
echo ""
echo "=== Summary ==="
echo " FULL archive: $FULL_SIZE"
echo " MINIMAL archive: $MINIMAL_SIZE"
echo ""
echo "Upload these files to your Gitea instance."
echo "Then update flake.nix to use fetchurl instead of local path."