viable/strict/1790898650: [CUDA][LinAlg] Nopiv LU - better heuristics (#193004)
- PyTorch 近 90 天出现 1779 次
- PyTorch 近 90 天第 1757 次发版
- 上一次：同一天稍早 · viable/strict/1790897185
发生了什么
Without these we inflict up to 7x regression on inputs with low batch count, and up to 2x for high-batched inputs. With about 540 points (gathered for A100, H100, GB200), this heuristic gets about 98% right. This is a prep work for LU improvements without the pivoting where MAGMA still outperforms cuBLAS/cuSOLVER. Pull Request resolved: #193004 Approved by: https://github.com/eqy
摘要按规则整理自下方来源原文