Merge pull request #16160 from rahulghangas/LocalDiagFix
Change local diag to support domains with non-uniform dimensions
Add diag support for matrices with different range offsets/strides.
- [x] Add further tests for non-square matrices
[Contributed by @rahulghangas]
[Reviewed by @ben-albrecht]