← 返回事件
持续讨论AI发版

trunk/5d1d994be2d895f42bcfda682358b416ceb902c2: [torchfuzz] Stop the matmul family emitting a redundant cast (#197064)

图:PyTorch Releases

发生了什么

Summary: MMOperator , AddmmOperator , BmmOperator and MatmulOperator all end codegen by casting every operand to the output dtype: out = torch.matmul(a.to(torch.float32), b.to(torch.float32)) That cast can never convert anything. fuzz_inputs_specs assigns input dtypes from _get_compatible_dtype , which is defined once on the base class as return [output_dtype, output_dtype] and is never overridden, so every input is…

摘要按规则整理自下方来源原文

为什么在扩散

来源