Prhub

#27815 [AMD] Update test_aiter_allgather_amd.py data types alignment between benchmark aiter and custom all-reduce kernel

原始 PR 作者 kangwangamd 合并时间 2026-06-17 15:48 文件变更 1 提交数 2 评论 2 代码增减 +0 / -7

执行摘要

AMD 测试删除了不支持的整型列

test_aiter_allgather_amd.py 测试在 AMD CI 中因指定了底层 all-reduce kernel 不支持的整型 dtype 而失败,这些 dtype 本就不应该被测试覆盖。

简单 bugfix,无需精读,但展示了测试应基于底层能力边界设计的原则。

实现拆解

  1. 修改 test/registered/ops/test_aiter_allgather_amd.py 中的 dtype_names 列表:删除 7 个整型 dtype(uint64_t, int64_t, uint32_t, int32_t, int16_t, uint8_t, int8_t),只保留 float32, float16, bfloat16。
  2. 同步更新了注释和说明。
文件 模块 状态 重要度
test/registered/ops/test_aiter_allgather_amd.py AMD AllGather modified 3.59

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

无风险:仅测试层面移除不支持的 dtype,不影响任何 kernel 或模型逻辑。

仅影响 AMD CI 中该测试的通过情况,消除假阳性失败。

测试覆盖调整

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论