viable/strict/1781045526: [MPS] Metal cumsum cumprod kernels (#185609)
🔒
https://github.com
«Fixes #154881 and #184844
PR is a bit of a big one so I'll try to summarize below:
Each row is an example tensor (scan dim + dtype) and the kernel it lands on. 1M = a long axis, the specialized long-scan kernels need ax...»
Automatische Weiterleitung...
1.5s