💾 viable/strict/1781045526: [MPS] Metal cumsum cumprod kernels (#185609)
Nachrichtenbereich: 💾 Downloads
🔗 Quelle: 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... [Weiterlesen]