diff --git a/mlir/unittests/Dialect/SparseTensor/MergerTest.cpp b/mlir/unittests/Dialect/SparseTensor/MergerTest.cpp --- a/mlir/unittests/Dialect/SparseTensor/MergerTest.cpp +++ b/mlir/unittests/Dialect/SparseTensor/MergerTest.cpp @@ -260,8 +260,6 @@ case kBitCast: case kBinaryBranch: case kUnary: - case kShlI: - case kBinary: return compareExpression(tensorExp.children.e0, pattern->e0); // Binary operations. case kMulF: @@ -282,6 +280,8 @@ case kXorI: case kShrS: case kShrU: + case kShlI: + case kBinary: return compareExpression(tensorExp.children.e0, pattern->e0) && compareExpression(tensorExp.children.e1, pattern->e1); }