[mlir][sparse] replace zero yield generic op with copy in allocation
authorAart Bik <ajcbik@google.com>
Wed, 3 Aug 2022 22:15:42 +0000 (15:15 -0700)
committerAart Bik <ajcbik@google.com>
Thu, 4 Aug 2022 16:33:57 +0000 (09:33 -0700)
commitc7bb69bc7546887b98c5cc3e5c50318f85b56eaf
tree4151a67e78c50cf3a69c6815399cf4f3eb9345ea
parent213c21fe10bb1bc72efddc5828761f445f16f28c
[mlir][sparse] replace zero yield generic op with copy in allocation

This prepares patterns that sometimes are generated by the front-end
and would prohibit fusion of SDDMM flavored kernels.

Reviewed By: springerm

Differential Revision: https://reviews.llvm.org/D131126
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
mlir/test/Dialect/SparseTensor/sparse_sddmm.mlir [changed mode: 0644->0755]