#36092 [ROCm] Fix AITER ops fake impl and minor bugs
原始 PR · 作者 ChuanLi1101 · 合并时间 2026-04-10 08:56
修复ROCm平台AITER算子fake实现返回None、静态方法参数错误和错误信息格式问题。
该PR值得ROCm平台开发者或关注AITER算子实现的工程师精读,尤其是fake实现的设计,展示了如何为自定义算子提供兼容torch.compile的元实现。关注点:fake实现如何模拟真实算子的输出张量形状和类型,这是支持torch.compile的关键模式。
参与讨论