c503dc1b8a
Add support for vectorization for linalg.generic representing element-wise ops. Those are converted to transfer_read + vector ops + transfer_write. Also re-organize the vectorization tests to be together. Implementation derived from the work of @burmako, @agrue and @fedelebron. Differential Revision: https://reviews.llvm.org/D92540
28 lines
1.3 KiB
MLIR
28 lines
1.3 KiB
MLIR
// RUN: mlir-opt %s -test-linalg-transform-patterns=test-matmul-to-vector-patterns-tile-1d | FileCheck %s
|
|
// RUN: mlir-opt %s -test-linalg-transform-patterns=test-matmul-to-vector-patterns-tile-2d | FileCheck %s
|
|
|
|
func @matmul(%A: memref<1584x1584xf32, offset: 0, strides: [1584, 1]>,
|
|
%B: memref<1584x1584xf32, offset: 0, strides: [1584, 1]>,
|
|
%C: memref<1584x1584xf32, offset: 0, strides: [1584, 1]>) {
|
|
linalg.matmul {__internal_linalg_transform__ = "START"}
|
|
ins(%A, %B: memref<1584x1584xf32, offset: 0, strides: [1584, 1]>,
|
|
memref<1584x1584xf32, offset: 0, strides: [1584, 1]>)
|
|
outs(%C: memref<1584x1584xf32, offset: 0, strides: [1584, 1]>)
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL:func @matmul
|
|
// CHECK: vector.transfer_write {{.*}} : vector<8x16xf32>, memref<8x16xf32>
|
|
// CHECK: vector.transfer_write {{.*}} : vector<16x12xf32>, memref<16x12xf32>
|
|
// CHECK: vector.transfer_write {{.*}} : vector<8x12xf32>, memref<8x12xf32>
|
|
//
|
|
// CHECK: linalg.copy
|
|
// CHECK: linalg.copy
|
|
// CHECK: linalg.copy
|
|
//
|
|
// CHECK: vector.contract
|
|
// CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"]
|
|
// CHECK-SAME: : vector<8x16xf32>, vector<16x12xf32> into vector<8x12xf32>
|
|
//
|
|
// CHECK: linalg.copy
|