From 5799f843a22029bd51d45edcd773e3c8662a0a08 Mon Sep 17 00:00:00 2001 From: Aart Bik Date: Tue, 24 May 2022 11:49:47 -0700 Subject: [PATCH] [mlir][sparse] add new complex ops to reduction recognition Reviewed By: bixia Differential Revision: https://reviews.llvm.org/D126318 --- mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp index 1a13eec..6453cf9 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: -- 2.7.4