diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp --- a/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp +++ b/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp @@ -378,11 +378,14 @@ static Reduction getReduction(Kind kind) { switch (kind) { case Kind::kAddF: + case Kind::kAddC: case Kind::kAddI: case Kind::kSubF: + case Kind::kSubC: case Kind::kSubI: return kSum; case Kind::kMulF: + case Kind::kMulC: case Kind::kMulI: return kProduct; case Kind::kAndI: