Skip to content

benchmark: 更新 7_MoeGatingTopKSoftmax / 9_TopKTopP 参考实现,增加偏态数据生成 - #171

Open
serlar wants to merge 1 commit into
Just-it:br_430from
serlar:br_430
Open

benchmark: 更新 7_MoeGatingTopKSoftmax / 9_TopKTopP 参考实现,增加偏态数据生成#171
serlar wants to merge 1 commit into
Just-it:br_430from
serlar:br_430

Conversation

@serlar

@serlar serlar commented May 4, 2026

Copy link
Copy Markdown

变更说明

更新 7_MoeGatingTopKSoftmax 和 9_TopKTopP 的参考实现(docstring 内 PyTorch 参考代码),增加偏态数据生成逻辑,使输入分布更接近真实 MoE gating logits 分布,提升参考实现的可信度。

变更类型

  • 新增算子 / Skill
  • 性能优化
  • Bug 修复
  • Benchmark(新增 case / 修改评测逻辑)
  • Agent / 框架改动
  • 文档 / 基础设施

影响范围

  • Triton 侧
  • AscendC 侧
  • 共享(router / benchmark-scheduler)

性能数据(涉及算子生成/优化时必填)

Benchmark 评测(与 BASELINE.md 对比)

指标 BASELINE 本次评测
编译通过数 - -
精度通过数 - -
平均 Speedup - -

测试环境

  • 设备型号:
  • CANN 版本:
  • PyTorch 版本:

冒烟测试(涉及算子生成/框架改动时必填)

  • Triton 通路:✅ / ❌(失败原因:)
  • AscendC 通路:✅ / ❌(失败原因:)

验证清单

  • 双通路冒烟测试通过
  • 通过率不退化(编译、精度均 >= BASELINE)
  • 平均 Speedup 不退化(>= BASELINE × 0.95)
  • 性能优化类:已跑全量评测、逐任务无退化、至少 1 个提升 >= 5%

退化说明(如有通过率下跌)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant