[mlir][sparse] fix crash when sparsifying broadcast operations.
authorPeiming Liu <peiming@google.com>
Thu, 27 Oct 2022 17:12:20 +0000 (17:12 +0000)
committerPeiming Liu <peiming@google.com>
Thu, 27 Oct 2022 17:43:22 +0000 (17:43 +0000)
commit015bc3464d7546b48b8edf4fd452143fa77a96a4
treee59e218b434fa798e687a749be85b04bcfcfb477
parent8fa32a74cf1ddb44812505f30f2d970cd507adbb
[mlir][sparse] fix crash when sparsifying broadcast operations.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D136866
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.h
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/test/Dialect/SparseTensor/sparse_broadcast.mlir [new file with mode: 0644]