viable/strict/1788555000: [MPS] Fix svd_jacobi Metal compiler crash (#195949)
Root cause: The AGX back-end could not handle a non-constant barrier scope produced by a ternary expression, while the Metal front-end accepted it. This caused a fatal error in the MTLCompilerService and a crash during pipeline creation for the affected SVD/lstsq paths.
Fix implemented: The patch ensures stable kernel compilation by aligning barrier scope handling with backend capabilities and introduces regression tests that force the native kernel paths for svd/svdvals and batched lstsq to prevent regressions.
Impact: Users should observe the resumption of SVD-related operations without the prior Metal compilation crash on AGX devices, with improved test coverage around these code paths.
Why it matters: Stabilizes SVD and lstsq operations on AGX Metal by preventing a compiler crash; improves confidence that changes won’t reintroduce the issue through regressions.
AI-assisted brief
official source
Add a comment
No account required