diff options
Diffstat (limited to 'mlir/lib')
-rw-r--r-- | mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp b/mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp index f844987e5a70..6ef79f868f8a 100644 --- a/mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp +++ b/mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp @@ -807,7 +807,7 @@ bool Merger::maybeZero(unsigned e) const { if (auto c = tensorExps[e].val.getDefiningOp<complex::ConstantOp>()) { ArrayAttr arrayAttr = c.getValue(); return arrayAttr[0].cast<FloatAttr>().getValue().isZero() && - arrayAttr[0].cast<FloatAttr>().getValue().isZero(); + arrayAttr[1].cast<FloatAttr>().getValue().isZero(); } if (auto c = tensorExps[e].val.getDefiningOp<arith::ConstantIntOp>()) return c.value() == 0; |