[test] Check depthwise conv is vectorized in test (NFC) (iree-org#18225)

Quick follow-up to iree-org#18190, as I realized the test may also pass if the conv fails to vectorize. Signed-off-by: Benjamin Maxwell <[email protected]>
nod-ai · Aug 15, 2024 · b144e90 · b144e90
1 parent 53a7bc4
commit b144e90
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/Common/test/generic_vectorization.mlir b/compiler/src/iree/compiler/Codegen/Common/test/generic_vectorization.mlir
@@ -503,3 +503,6 @@ func.func @depthwise_conv_fold_away_masking(%arg0: tensor<1x68x120x96xf32>, %arg
 // CHECK-MASK-LABEL: func.func @depthwise_conv_fold_away_masking
 // CHECK-MASK-NOT: vector.create_mask
 // CHECK-MASK-NOT: vector.constant_mask
+// CHECK-MASK:     vector.fma
+// CHECK-MASK-NOT: vector.create_mask
+// CHECK-MASK-NOT: vector.constant_mask