From ba270da07ba2f36b4e11b4e56ab3da01cfa97457 Mon Sep 17 00:00:00 2001 From: Benjamin Maxwell Date: Thu, 15 Aug 2024 09:35:07 +0000 Subject: [PATCH] [test] Check depthwise conv is vectorized in test (NFC) Quick follow-up to #18190, as I realized the test may also pass if the conv fails to vectorize. --- .../compiler/Codegen/Common/test/generic_vectorization.mlir | 3 +++ 1 file changed, 3 insertions(+) diff --git a/compiler/src/iree/compiler/Codegen/Common/test/generic_vectorization.mlir b/compiler/src/iree/compiler/Codegen/Common/test/generic_vectorization.mlir index 38e0f5ea2a3f0..a1e98b3d2324f 100644 --- 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