Ascend/run-mindspeed-llm-test

MindSpeed-LLM 测试执行器

执行MindSpeed-LLM项目的单元测试用例。

运行环境

运行mindspeed-llm测试用例时请严格按照以下运行环境的要求进行

• Docker容器：q4_master
• 默认环境配置：'source /root/.bashrc'
• 默认工作目录：当前项目工作目录，记为WORK_DIR
• NPU环境配置：'source /usr/local/Ascend/ascend-toolkit/set_env.sh && source /usr/local/Ascend/nnal/atb/set_env.sh'

输入

用户输入执行脚本（如run_coverage.sh脚本）或者指定测试用例目录

执行流程

1.进入docker容器

docker exec -it q4_master /bin/bash

2.执行默认环境配置

source /root/.bashrc

3.进入项目工作目录

cd ${WORK_DIR}

4.运行任务 执行测试命令或脚本

# 执行全量代码覆盖率扫描任务
bash tests/run_coverage.sh all

# 执行pipeline所有用例
bash tests/pipeline/pipe_run.sh

# 执行ut测试用例
pytest -s tests/ut/inference/test_inference.py

5.分析运行结果

输出

输出测试报告或覆盖率报告，以markdown文件的形式至TEST目录下

示例

执行全量代码覆盖率扫描任务

docker exec -it q4_master bash -c "source /root/.bashrc && cd ${WORK_DIR} && bash tests/run_coverage.sh all"

运行pipeline所有用例

docker exec -it q4_master bash -c "source /root/.bashrc && cd ${WORK_DIR} && bash tests/pipeline/pipe_run.sh"

运行UT测试用例

# 运行一个ut测试用例文件中的所有用例
docker exec -it q4_master bash -c "source /root/.bashrc && cd ${WORK_DIR} && pytest -s tests/pipeline/ut/model_module/test_attention.py"

# 运行一个ut测试用例中的特定测试用例
docker exec -it q4_master bash -c "source /root/.bashrc && cd ${WORK_DIR} && pytest -s tests/pipeline/ut/model_module/test_attention.py::TestAttention::test_alibi_seq8192_bs2_bf16"

运行ST测试用例

docker exec -it q4_master bash -c "source /root/.bashrc && cd ${WORK_DIR} && bash tests/pipeline/st/llama2/llama2_tp8_pp1_coc_ptd.sh"