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

viable/strict/1788067484: [CUDA] Add non-overlapping fast path for avg_pool2d backward (#191086)

图:PyTorch Releases

发生了什么

Summary Adds a dedicated CUDA backward kernel for avg_pool2d when the pooling windows tile the input exactly ( stride == kernel_size , no padding, no ceil_mode ) -- the common downsampling case such as avg_pool2d(2, 2) . In that regime each input element maps to exactly one output window with a constant divisor, so the general kernel's per-element window scan and per-window bound/divisor recomputation collapse to a…

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

为什么在扩散

来源