viable/strict/1788130193: [BE] [MPS] Use a typed struct for LU scratch space (#195297)
- PyTorch 近 90 天出现 283 次
- PyTorch 近 90 天第 280 次发版
- 上一次:同一天稍早 · viable/strict/1788128625
发生了什么
Introduce c10::metal::complex alias to either c10::complex or metal::vec Introduce LUStreamScratch that holds float, uint and T scratch spaces for various stages of LU decomposition That allows one to kill pointer arithmetic on device and allocate it as sizeof(LUStreamScratch) rather than some complex dance of magic constants 🤖 Generated with Claude Code Pull Request resolved: #195297 Approved by: https://github.co…
摘要按规则整理自下方来源原文