| commit | f16cb0eade08035fea5e8310bd4a64c8f286c929 | |
| author | Aart Bik <39774503+aartbik@users.noreply.github.com> | |
| Thu, 19 Oct 2023 01:00:56 +0000 (18 18:00 -0700) | ||
| committer | GitHub <noreply@github.com> | |
| Thu, 19 Oct 2023 01:00:56 +0000 (18 18:00 -0700) | ||
| tree | 8f54892bdcd3e5110d69c0e5b4a100ae0e53381b | treesnapshot (tar.gz zip) |
| parent | e103515cedff16935f8f84e86cf78316cf33c220 | commitdiff |
[mlir][sparse] connect MapRef's lvl2dim with latest AffineMap computation (#69540)
This makes sure
- GEN MAP dim=2 lvl=4
(d0, d1) -> (d0 floordiv 2, d1 floordiv 2, d0 mod 2, d1 mod 2)
--
(d0, d1, d2, d3) -> (d0 * 2 + d2, d1 * 2 + d3)
is indeed encoded as
MAP-REF (dim=2, lvl=4) isperm=0
d2l = [ d0/2 d1/2 d0%2 d1%2 ]
ld2 = [ l2+2*l0 l3+2*l1 ]
This makes sure
- GEN MAP dim=2 lvl=4
(d0, d1) -> (d0 floordiv 2, d1 floordiv 2, d0 mod 2, d1 mod 2)
--
(d0, d1, d2, d3) -> (d0 * 2 + d2, d1 * 2 + d3)
is indeed encoded as
MAP-REF (dim=2, lvl=4) isperm=0
d2l = [ d0/2 d1/2 d0%2 d1%2 ]
ld2 = [ l2+2*l0 l3+2*l1 ]
| mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp | diffblobblamehistory |