[all-commits] [llvm/llvm-project] f16cb0: [mlir][sparse] connect MapRef's lvl2dim with lates...

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f16cb0eade08035fea5e8310bd4a64c8f286c929
      https://github.com/llvm/llvm-project/commit/f16cb0eade08035fea5e8310bd4a64c8f286c929
  Author: Aart Bik <39774503+aartbik at users.noreply.github.com>
  Date:   2023-10-18 (Wed, 18 Oct 2023)

  Changed paths:
    M mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp

  Log Message:
  -----------
  [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 ]