[mlir][sparse] make test for block sparsity more robust (#74798)
commit3d3e46cc4db9dd32edc82b7029fb694d5d0316de
authorAart Bik <39774503+aartbik@users.noreply.github.com>
Fri, 8 Dec 2023 19:50:10 +0000 (8 11:50 -0800)
committerGitHub <noreply@github.com>
Fri, 8 Dec 2023 19:50:10 +0000 (8 11:50 -0800)
tree5051e22cf214c897207b51e6b3198db50de99a8e
parent944e031e36d9515b68b320f611edfc97d5460259
[mlir][sparse] make test for block sparsity more robust (#74798)

For BSR and convolutions, we encounter

(d0, d1, d2, d3) -> ((d0 + d2) floordiv 2, (d1 + d3) floordiv 2, (d0 +
d2) mod 2, (d1 + d3) mod 2)

which crashed the current test. Note that an actual test and working
code is still to follow (since we need to fix a few other things first)
mlir/lib/Dialect/SparseTensor/IR/SparseTensorDialect.cpp