trunk/769b13f46a24d108e74a807bc3bdb5cc8b478758: Add tiled CUDA kernel for dense 2D transpose copies (#194310)
- PyTorch 近 90 天出现 1410 次
- PyTorch 近 90 天第 1392 次发版
- 上一次:同一天稍早 · trunk/98950cfd474df1053b0c1f7d686de13302b20e92
发生了什么
Addresses #113560 . t().contiguous() on CUDA goes through the generic TensorIterator elementwise kernel, which can't coalesce both the read and the write of a transpose. This adds a shared-memory tiled kernel for copies that TensorIterator reduces to a 2D transpose, including sources with a row pitch larger than the row (narrowed or strided outer dims). Kernel 128-bit global loads and stores for aligned 1-, 2- and 4…
摘要按规则整理自下方来源原文