为 Fused MoE 基准测试增加报告输出 #59

Mengz · 2026-06-26T10:04:32+08:00

Mengz commented

2026-06-26 10:04:32 +08:00

First-time contributor

现有 Fused MoE Python benchmark 只在控制台打印结果，每次在不同箱型、不同 warmup 和 iters 下对比数据时，都需要额外手工整理环境信息、配置参数和基准输出，很难留下可回溯的记录。

这个提交给 benchmark 补上了 CSV、JSON 和运行报告所需的核心数据，同时增加 output-dir、case-filter、设备同步和环境采集能力，让短跑验证和长跑对比都能直接落地归档。

已在沐曦 GPU 环境用 warmup=2、iters=5 进行实际验证，fused_moe_i8_tn_topk1、topk2、topk3 都能通过正确性检查并输出 avg_ms 与 TOPS，生成的 CSV/JSON 报告已作为本地测试记录保存。

现有 Fused MoE Python benchmark 只在控制台打印结果，每次在不同箱型、不同 warmup 和 iters 下对比数据时，都需要额外手工整理环境信息、配置参数和基准输出，很难留下可回溯的记录。这个提交给 benchmark 补上了 CSV、JSON 和运行报告所需的核心数据，同时增加 output-dir、case-filter、设备同步和环境采集能力，让短跑验证和长跑对比都能直接落地归档。已在沐曦 GPU 环境用 warmup=2、iters=5 进行实际验证，fused_moe_i8_tn_topk1、topk2、topk3 都能通过正确性检查并输出 avg_ms 与 TOPS，生成的 CSV/JSON 报告已作为本地测试记录保存。

Mengz added 1 commit 2026-06-26 10:04:32 +08:00

aab07de35f feat: write Fused MoE benchmark reports

This pull request has changes conflicting with the target branch.

基于AI Agent开发范式的国产GPU大模型推理算子库优化/baselines/fused_moe/standalone/fused_moe_i8_tn/python/benchmark_fused_moe_i8_tn.py

You can also view command line instructions.

Step 1:

From your project repository, check out a new branch and test the changes.

git checkout -b Mengz-mengz/fused-moe-benchmark-report master

git pull mengz/fused-moe-benchmark-report

Step 2:

Merge the changes and update on Gitea.

git checkout master

git merge --no-ff Mengz-mengz/fused-moe-benchmark-report

git push origin master

Sign in to join this conversation.

No reviewers

No Label

No Milestone

No project

No Assignees

1 Participants

Notifications

Due Date

The due date is invalid or out of range. Please use the format 'yyyy-mm-dd'.

No due date set.

Dependencies

No dependencies set.

Reference: metax-maca/op_optimization#59