[Mlir-commits] [mlir] 5799f84 - [mlir][sparse] add new complex ops to reduction recognition

[Aart Bik]

Author: Aart Bik
Date: 2022-05-24T15:00:56-07:00
New Revision: 5799f843a22029bd51d45edcd773e3c8662a0a08

URL: https://github.com/llvm/llvm-project/commit/5799f843a22029bd51d45edcd773e3c8662a0a08
DIFF: https://github.com/llvm/llvm-project/commit/5799f843a22029bd51d45edcd773e3c8662a0a08.diff

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

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D126318

Added: 
    

Modified: 
    mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
index 1a13eecb846ea..6453cf99e09fe 100644
--- 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: