Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-01-09 17:43:57 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source 
Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D126318
This commit is contained in:
Aart Bik 2022-05-24 11:49:47 -07:00
parent d2ee2c9c8d
commit 5799f843a2
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
View File

@ -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: