#51457 [Test] Add ROCm AITER FP8 MLA prefill accuracy test
原始 PR · 作者 aarushjain29 · 合并时间 2026-08-09 08:42
为 gfx950 新增 AITER FP8 MLA prefill 精度测试
值得精读。该 PR 展示了一种轻量级 kernel 精度测试模式:用 `object.__new__` 绕过 `__init__`、以 `SimpleNamespace` 代替 dataclass,在无需初始化完整模型的前提下驱动真实的 metadata builder 与 impl 入口,避免测试与实现脱节。建议跟进 reviewer 的两个 follow-up:确认 device 字符串语义、对容差做数值验证;同时可参考 `tests/v1/attention/test_sparse_mla_backends.py` 等近期 MLA 测试的演进,评估是否可以把类似模式推广到更多 ROCm 内核路径。
参与讨论