[mlir][sparse] add new complex ops to reduction recognition

Reviewed By: bixia Differential Revision: https://reviews.llvm.org/D126318
2022-05-24 11:49:47 -07:00 · 2022-05-24 11:49:47 -07:00 · 5799f843a2
parent d2ee2c9c8d
commit 5799f843a2
1 changed files with 3 additions and 0 deletions
--- a/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
+++ b/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
@ -378,11 +378,14 @@ static vector::CombiningKind getCombiningKind(Reduction kind) {
 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: