Skip to content

Commit

Permalink
Revert "[mlir] Add vectorization tests for linalg.map,reduce,transpose."
Browse files Browse the repository at this point in the history
This reverts commit 2f88268.

I will take a look what's happening with vectorization.mlir and why it
works on my machine and not upstream. Reverting for now.
  • Loading branch information
pifon2a committed Oct 25, 2022
1 parent d03ee70 commit c34de60
Showing 1 changed file with 0 additions and 68 deletions.
68 changes: 0 additions & 68 deletions mlir/test/Dialect/Linalg/vectorization.mlir
Original file line number Diff line number Diff line change
Expand Up @@ -1522,71 +1522,3 @@ transform.sequence failures(propagate) {
%1 = get_closest_isolated_parent %0 : (!pdl.operation) -> !pdl.operation
%2 = transform.structured.vectorize %1
}

// -----

func.func @vectorize_map(%arg0: memref<64xf32>,
%arg1: memref<64xf32>, %arg2: memref<64xf32>) {
linalg.map ins(%arg0, %arg1 : memref<64xf32>, memref<64xf32>)
outs(%arg2 : memref<64xf32>)
(%in: f32, %in_0: f32) {
%0 = arith.addf %in, %in_0 : f32
linalg.yield %0 : f32
}
return
}
// CHECK-LABEL: func @vectorize_map
// CHECK: %[[LHS:.*]] = vector.transfer_read
// CHECK-NEXT: %[[RHS:.*]] = vector.transfer_read
// CHECK-NEXT: arith.addf %[[LHS]], %[[RHS]] : vector<64xf32>

transform.sequence failures(propagate) {
^bb1(%arg1: !pdl.operation):
%0 = transform.structured.match ops{["linalg.map"]} in %arg1
%1 = get_closest_isolated_parent %0 : (!pdl.operation) -> !pdl.operation
%2 = transform.structured.vectorize %1
}

// -----

func.func @vectorize_transpose(%arg0: memref<16x32x64xf32>,
%arg1: memref<32x64x16xf32>) {
linalg.transpose ins(%arg0 : memref<16x32x64xf32>)
outs(%arg1 : memref<32x64x16xf32>) permutation = [1, 2, 0]
return
}
// CHECK-LABEL: func @vectorize_transpose
// CHECK: %[[FIRST:.*]] = vector.transpose
// CHECK-SAME: [2, 0, 1] : vector<32x64x16xf32> to vector<16x32x64xf32>
// CHECK-NEXT: vector.transpose %[[FIRST]]
// CHECK-SAME: [1, 2, 0] : vector<16x32x64xf32> to vector<32x64x16xf32>

transform.sequence failures(propagate) {
^bb1(%arg1: !pdl.operation):
%0 = transform.structured.match ops{["linalg.transpose"]} in %arg1
%1 = get_closest_isolated_parent %0 : (!pdl.operation) -> !pdl.operation
%2 = transform.structured.vectorize %1
}

// -----

func.func @vectorize_reduce(%arg0: memref<16x32x64xf32>,
%arg1: memref<16x64xf32>) {
linalg.reduce ins(%arg0 : memref<16x32x64xf32>)
outs(%arg1 : memref<16x64xf32>) dimensions = [1]
(%in: f32, %init: f32) {
%0 = arith.addf %in, %init : f32
linalg.yield %0 : f32
}
return
}
// CHECK-LABEL: func @vectorize_reduce
// CHECK: vector.multi_reduction <add>
// CHECK-SAME: : vector<16x32x64xf32> to vector<16x64xf32>

transform.sequence failures(propagate) {
^bb1(%arg1: !pdl.operation):
%0 = transform.structured.match ops{["linalg.reduce"]} in %arg1
%1 = get_closest_isolated_parent %0 : (!pdl.operation) -> !pdl.operation
%2 = transform.structured.vectorize %1
}

0 comments on commit c34de60

Please sign in to comment.