Skip to content

Commit c84b7c0

Browse files
committed
fixup! [mlir][linalg][nfc] Move vectorization tests
* In vectorize-convolution.mlir, switch from `-test-linalg-transform-patterns=test-linalg-to-vector-patterns` to `transform.structured.vectorize_children_and_apply_patterns` (for consistency with other tests for vectorization). * Move + rename the remaining vectorization tests. * Add comments.
1 parent 14cf58d commit c84b7c0

File tree

5 files changed

+252
-7
lines changed

5 files changed

+252
-7
lines changed

mlir/test/Dialect/Linalg/vectorize-convolution-flatten.mlir renamed to mlir/test/Dialect/Linalg/vectorization/convolution-with-patterns-flatten.mlir

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,11 @@
11
// RUN: mlir-opt -split-input-file -transform-interpreter %s | FileCheck %s
22

3+
///----------------------------------------------------------------------------------------
4+
/// Tests for vectorizing depthwise convolutions (with patterns) with the
5+
/// flattening of the channel dim enabled. This approach is beneficial when the
6+
/// number of channel dimensions is low.
7+
///----------------------------------------------------------------------------------------
8+
39
func.func @depthwise_conv1d_nwc_wc_1x8x3xi8_tensor(%input: tensor<1x8x3xi8>,
410
%filter: tensor<1x3xi8>,
511
%output: tensor<1x8x3xi8>) -> (tensor<1x8x3xi8>) {

0 commit comments

Comments
 (0)