https://github.com/kuhar approved this pull request. LGTM. May be worth adding explicit static_casts: in `getRows` and `getColumns` and an assertion that none of the dims are dynamic. https://github.com/llvm/llvm-project/pull/142784