bbf362b784
The entire compiler now looks at structural properties of the function (e.g. does it have one block, does it contain an if/for stmt, etc) so the only thing holding up this difference is round tripping through the parser/printer syntax. Removing this shrinks the compile by ~140LOC. This is step 31/n towards merging instructions and statements. The last step is updating the docs, which I will do as a separate patch in order to split it from this mostly mechanical patch. PiperOrigin-RevId: 227540453
37 lines
2.0 KiB
MLIR
37 lines
2.0 KiB
MLIR
// RUN: mlir-opt %s -vectorizer-test -vector-shape-ratio 4 -vector-shape-ratio 8 | FileCheck %s
|
|
// RUN: mlir-opt %s -vectorizer-test -vector-shape-ratio 2 -vector-shape-ratio 5 -vector-shape-ratio 2 | FileCheck %s -check-prefix=TEST-3x4x5x8
|
|
|
|
func @vector_add_2d(%arg0: index, %arg1: index) -> f32 {
|
|
// Nothing should be matched in this first block.
|
|
// CHECK-NOT:matched: {{.*}} = alloc{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = constant 0{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = constant 1{{.*}}
|
|
%0 = alloc(%arg0, %arg1) : memref<?x?xf32>
|
|
%1 = alloc(%arg0, %arg1) : memref<?x?xf32>
|
|
%2 = alloc(%arg0, %arg1) : memref<?x?xf32>
|
|
%c0 = constant 0 : index
|
|
%cst = constant 1.000000e+00 : f32
|
|
|
|
// CHECK:matched: {{.*}} constant splat{{.*}} with shape ratio: 2, 32
|
|
%cst_1 = constant splat<vector<8x256xf32>, 1.000000e+00> : vector<8x256xf32>
|
|
// CHECK:matched: {{.*}} constant splat{{.*}} with shape ratio: 1, 3, 7, 2, 1
|
|
%cst_a = constant splat<vector<1x3x7x8x8xf32>, 1.000000e+00> : vector<1x3x7x8x8xf32>
|
|
// CHECK-NOT:matched: {{.*}} constant splat{{.*}} with shape ratio: 1, 3, 7, 1{{.*}}
|
|
%cst_b = constant splat<vector<1x3x7x4x4xf32>, 1.000000e+00> : vector<1x3x7x8x8xf32>
|
|
// TEST-3x4x5x8:matched: {{.*}} constant splat{{.*}} with shape ratio: 3, 2, 1, 4
|
|
%cst_c = constant splat<vector<3x4x5x8xf32>, 1.000000e+00> : vector<3x4x5x8xf32>
|
|
// TEST-3x4x4x8-NOT:matched: {{.*}} constant splat{{.*}} with shape ratio{{.*}}
|
|
%cst_d = constant splat<vector<3x4x4x8xf32>, 1.000000e+00> : vector<3x4x4x8xf32>
|
|
// TEST-3x4x4x8:matched: {{.*}} constant splat{{.*}} with shape ratio: 1, 1, 2, 16
|
|
%cst_e = constant splat<vector<1x2x10x32xf32>, 1.000000e+00> : vector<1x2x10x32xf32>
|
|
|
|
// Nothing should be matched in this last block.
|
|
// CHECK-NOT:matched: {{.*}} = constant 7{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = constant 42{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = load{{.*}}
|
|
// CHECK-NOT:matched: return {{.*}}
|
|
%c7 = constant 7 : index
|
|
%c42 = constant 42 : index
|
|
%9 = load %2[%c7, %c42] : memref<?x?xf32>
|
|
return %9 : f32
|
|
} |