SuperAPI AI 工具集

mcore-testing

测试与质量

查看项目

更新于 10-16 02:34

技能介绍

用途:这是 Megatron-LM 的测试指南,说明其测试体系如何组织、运行与维护,面向需要编写或调试测试的开发者。

主要能力:覆盖测试目录布局(unit_tests、functional_tests、test_utils)、recipe YAML 结构、单元测试与功能测试的添加和运行、golden values 管理、marker 过滤以及 CI 执行方式。

使用方式:单元测试全部初始化 torch.distributed 组,需 GPU 并通过 torch.distributed.run 运行,例如 uv run python -m torch.distributed.run --nproc-per-node 8 -m pytest -q tests/unit_tests;可用 -k 过滤名称、-m "not flaky and not flaky_in_dev" 排除不稳定用例、--experimental 包含实验用例。功能测试由 tests/functional_tests/shell_test_utils/run_ci_test.sh 驱动,recipe 位于 tests/test_utils/recipes/。

禁用而不删除:功能 recipe 将 scope 加 -broken 后缀(如 mr-github-broken);单元测试用 @pytest.mark.flaky_in_dev 或 @pytest.mark.flaky 跳过。

限制:golden values 保留 float32 全精度,不得四舍五入或手工编辑;确定性用例按位精确比较,旧文件按五位小数比较。GB200 任务需维护者触发。本地运行需注意 /opt/data 数据路径与 -s 导致的输出交错。