[test] Check depthwise conv is vectorized in test (NFC)

Quick follow-up to iree-org#18190, as I realized the test may also pass if the conv fails to vectorize.
MacDue · Aug 15, 2024 · ba270da · ba270da
1 parent 55759fa
commit ba270da
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: vector.fma
+// CHECK-MASK-NOT: vector.create_mask
+// CHECK-MASK-NOT: vector.constant_mask