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

ciflow/trunk/187952: Fix out= variant recompilations for torch.max/min/topk

图:PyTorch Releases

发生了什么

torch.max(x, dim, out=...), torch.min(..., out=...) and torch.topk(..., out=...) recompiled under torch.compile on every new input shape, while the functional variants specialize only once. With the default recompile_limit this also makes the function fall back to eager after 8 recompiles, losing compilation entirely. Root cause: the out overloads aten.max.dim_max, aten.min.dim_min and aten.topk.values had no Python…

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

为什么在扩散

来源