608a663c8e
Before the change `test-loop-fusion` and `affine-super-vectorizer-test`
options were in their own category. This was because they used the
standard llvm command line parsing with `llvm::cl::opt`. This PR moves
them over to the mlir `Pass::Option` class.
Before the change
```
$ mlir-opt --help
...
General options:
...
Compiler passes to run
Passes:
...
Pass Pipelines:
...
Generic Options:
....
affine-super-vectorizer-test options:
--backward-slicing
...
--vectorize-affine-loop-nest
test-loop-fusion options:
--test-loop-fusion-dependence-check
...
--test-loop-fusion-transformation
```
After the change
```
$ mlir-opt --help
...
General options:
...
Compiler passes to run
Passes:
...
--affine-super-vectorizer-test
--backward-slicing
...
--vectorize-affine-loop-nest
...
--test-loop-fusion options:
--test-loop-fusion-dependence-check
...
--test-loop-fusion-transformation
...
Pass Pipelines:
...
Generic Options:
...
```
---------
Signed-off-by: philass <plassen@groq.com>
63 lines
2.9 KiB
MLIR
63 lines
2.9 KiB
MLIR
// RUN: mlir-opt %s -affine-super-vectorizer-test="vector-shape-ratio=4,8" 2>&1 | FileCheck %s
|
|
// RUN: mlir-opt %s -affine-super-vectorizer-test="vector-shape-ratio=2,5,2" 2>&1 | FileCheck %s -check-prefix=TEST-3x4x5x8
|
|
// RUN: mlir-opt %s -affine-super-vectorizer-test=vectorize-affine-loop-nest 2>&1 | FileCheck %s -check-prefix=VECNEST
|
|
|
|
func.func @vector_add_2d(%arg0: index, %arg1: index) -> f32 {
|
|
// Nothing should be matched in this first block.
|
|
// CHECK-NOT:matched: {{.*}} = memref.alloc{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = arith.constant 0{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = arith.constant 1{{.*}}
|
|
%0 = memref.alloc(%arg0, %arg1) : memref<?x?xf32>
|
|
%1 = memref.alloc(%arg0, %arg1) : memref<?x?xf32>
|
|
%2 = memref.alloc(%arg0, %arg1) : memref<?x?xf32>
|
|
%c0 = arith.constant 0 : index
|
|
%cst = arith.constant 1.000000e+00 : f32
|
|
|
|
// CHECK:matched: {{.*}} arith.constant dense{{.*}} with shape ratio: 2, 32
|
|
%cst_1 = arith.constant dense<1.000000e+00> : vector<8x256xf32>
|
|
// CHECK:matched: {{.*}} arith.constant dense{{.*}} with shape ratio: 1, 3, 7, 2, 1
|
|
%cst_a = arith.constant dense<1.000000e+00> : vector<1x3x7x8x8xf32>
|
|
// CHECK-NOT:matched: {{.*}} arith.constant dense{{.*}} with shape ratio: 1, 3, 7, 1{{.*}}
|
|
%cst_b = arith.constant dense<1.000000e+00> : vector<1x3x7x4x4xf32>
|
|
// TEST-3x4x5x8:matched: {{.*}} arith.constant dense{{.*}} with shape ratio: 3, 2, 1, 4
|
|
%cst_c = arith.constant dense<1.000000e+00> : vector<3x4x5x8xf32>
|
|
// TEST-3x4x5x8-NOT:matched: {{.*}} arith.constant dense{{.*}} with shape ratio{{.*}}
|
|
%cst_d = arith.constant dense<1.000000e+00> : vector<3x4x4x8xf32>
|
|
// TEST-3x4x5x8:matched: {{.*}} arith.constant dense{{.*}} with shape ratio: 1, 1, 2, 16
|
|
%cst_e = arith.constant dense<1.000000e+00> : vector<1x2x10x32xf32>
|
|
|
|
// Nothing should be matched in this last block.
|
|
// CHECK-NOT:matched: {{.*}} = arith.constant 7{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = arith.constant 42{{.*}}
|
|
// CHECK-NOT:matched: {{.*}} = memref.load{{.*}}
|
|
// CHECK-NOT:matched: return {{.*}}
|
|
%c7 = arith.constant 7 : index
|
|
%c42 = arith.constant 42 : index
|
|
%9 = memref.load %2[%c7, %c42] : memref<?x?xf32>
|
|
return %9 : f32
|
|
}
|
|
|
|
// VECNEST-LABEL: func @double_loop_nest
|
|
func.func @double_loop_nest(%a: memref<20x30xf32>, %b: memref<20xf32>) {
|
|
|
|
affine.for %i = 0 to 20 {
|
|
%b_ld = affine.load %b[%i] : memref<20xf32>
|
|
affine.for %j = 0 to 30 {
|
|
%a_ld = affine.load %a[%i, %j] : memref<20x30xf32>
|
|
affine.store %a_ld, %a[%i, %j] : memref<20x30xf32>
|
|
}
|
|
affine.store %b_ld, %b[%i] : memref<20xf32>
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// VECNEST: affine.for %{{.*}} = 0 to 20 step 4 {
|
|
// VECNEST: vector.transfer_read
|
|
// VECNEST-NEXT: affine.for %{{.*}} = 0 to 30 {
|
|
// VECNEST: vector.transfer_read
|
|
// VECNEST-NEXT: vector.transfer_write %{{.*}}, %{{.*}}[%{{.*}}, %{{.*}}] {permutation_map = #{{.*}}}
|
|
// VECNEST-NEXT: }
|
|
// VECNEST-NEXT: vector.transfer_write
|
|
// VECNEST: }
|