File tree Expand file tree Collapse file tree 1 file changed +5
-7
lines changed
mlir/lib/Dialect/ArmNeon/Transforms Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Original file line number Diff line number Diff line change @@ -158,13 +158,11 @@ class VectorContractRewriter {
158158 PatternRewriter &rewriter) {
159159 // Check iterator types for matrix multiplication.
160160 SmallVector<vector::IteratorType> itTypes = op.getIteratorTypesArray ();
161- if (!((itTypes.size () == 3 &&
162- (itTypes[0 ] == vector::IteratorType::parallel &&
163- itTypes[1 ] == vector::IteratorType::parallel &&
164- itTypes[2 ] == vector::IteratorType::reduction)) ||
165- (itTypes.size () == 2 &&
166- (itTypes[0 ] == vector::IteratorType::parallel &&
167- itTypes[1 ] == vector::IteratorType::reduction))))
161+ if ((itTypes.size () != 3 || itTypes[0 ] != vector::IteratorType::parallel ||
162+ itTypes[1 ] != vector::IteratorType::parallel ||
163+ itTypes[2 ] != vector::IteratorType::reduction) &&
164+ (itTypes.size () != 2 || itTypes[0 ] != vector::IteratorType::parallel ||
165+ itTypes[1 ] != vector::IteratorType::reduction))
168166 return rewriter.notifyMatchFailure (
169167 op, " iterator types do not correspond to matrix multiplication" );
170168
You can’t perform that action at this time.
0 commit comments