7446151236
This allows individual passes to define options structs and for these options to be parsed per instance of the pass while building the pass pipeline from the command line provided textual specification.
The user can specify these per-instance pipeline options like so:
```
struct MyPassOptions : public PassOptions<MyPassOptions> {
Option<int> exampleOption{*this, "flag-name", llvm:🆑:desc("...")};
List<int> exampleListOption{*this, "list-flag-name", llvm:🆑:desc("...")};
};
static PassRegistration<MyPass, MyPassOptions> pass("my-pass", "description");
```
PiperOrigin-RevId: 273650140
134 lines
5.4 KiB
MLIR
134 lines
5.4 KiB
MLIR
// RUN: mlir-opt -test-extract-fixed-outer-loops='test-outer-loop-sizes=7' %s | FileCheck %s --check-prefixes=COMMON,TILE_7
|
|
// RUN: mlir-opt -test-extract-fixed-outer-loops='test-outer-loop-sizes=7,4' %s | FileCheck %s --check-prefixes=COMMON,TILE_74
|
|
|
|
// COMMON-LABEL: @rectangular
|
|
func @rectangular(%arg0: memref<?x?xf32>) {
|
|
%c2 = constant 2 : index
|
|
%c44 = constant 44 : index
|
|
%c1 = constant 1 : index
|
|
// Range of the original loop:
|
|
// (upper - lower + step - 1) / step
|
|
// where step is known to be %c1.
|
|
// COMMON: %[[diff:.*]] = subi %c44, %c2
|
|
// COMMON: %[[adjustment:.*]] = subi %c1, %c1_{{.*}}
|
|
// COMMON-NEXT: %[[diff_adj:.*]] = addi %[[diff]], %[[adjustment]]
|
|
// COMMON-NEXT: %[[range:.*]] = divis %[[diff_adj]], %c1
|
|
|
|
// Ceildiv to get the parametric tile size.
|
|
// COMMON: %[[sum:.*]] = addi %[[range]], %c6
|
|
// COMMON-NEXT: %[[size:.*]] = divis %[[sum]], %c7
|
|
// New outer step (original is %c1).
|
|
// COMMON-NEXT: %[[step:.*]] = muli %c1, %[[size]]
|
|
|
|
// Range of the second original loop
|
|
// (upper - lower + step - 1) / step
|
|
// where step is known to be %c2.
|
|
// TILE_74: %[[diff2:.*]] = subi %c44, %c1
|
|
// TILE_74: %[[adjustment2:.*]] = subi %c2, %c1_{{.*}}
|
|
// TILE_74-NEXT: %[[diff2_adj:.*]] = addi %[[diff2]], %[[adjustment2]]
|
|
// TILE_74-NEXT: %[[range2:.*]] = divis %[[diff2_adj]], %c2
|
|
|
|
// Ceildiv to get the parametric tile size for the second original loop.
|
|
// TILE_74: %[[sum2:.*]] = addi %[[range2]], %c3
|
|
// TILE_74-NEXT: %[[size2:.*]] = divis %[[sum2]], %c4
|
|
// New inner step (original is %c2).
|
|
// TILE_74-NEXT: %[[step2:.*]] = muli %c2, %[[size2]]
|
|
|
|
// Updated outer loop(s) use new steps.
|
|
// COMMON: loop.for %[[i:.*]] = %c2 to %c44 step %[[step]]
|
|
// TILE_74:loop.for %[[j:.*]] = %c1 to %c44 step %[[step2]]
|
|
loop.for %i = %c2 to %c44 step %c1 {
|
|
// Upper bound for the inner loop min(%i + %step, %c44).
|
|
// COMMON: %[[stepped:.*]] = addi %[[i]], %[[step]]
|
|
// COMMON-NEXT: cmpi "slt", %c44, %[[stepped]]
|
|
// COMMON-NEXT: %[[ub:.*]] = select {{.*}}, %c44, %[[stepped]]
|
|
//
|
|
// TILE_74: %[[stepped2:.*]] = addi %[[j]], %[[step2]]
|
|
// TILE_74-NEXT: cmpi "slt", %c44, %[[stepped2]]
|
|
// TILE_74-NEXT: %[[ub2:.*]] = select {{.*}}, %c44, %[[stepped2]]
|
|
|
|
// Created inner loop.
|
|
// COMMON:loop.for %[[ii:.*]] = %[[i]] to %[[ub:.*]] step %c1
|
|
|
|
// This loop is not modified in TILE_7 case.
|
|
// TILE_7: loop.for %[[j:.*]] = %c1 to %c44 step %c2
|
|
//
|
|
// But is modified in TILE_74 case.
|
|
// TILE_74:loop.for %[[jj:.*]] = %[[j]] to %[[ub2]] step %c2
|
|
loop.for %j = %c1 to %c44 step %c2 {
|
|
// The right iterator are used.
|
|
// TILE_7: load %arg0[%[[ii]], %[[j]]]
|
|
// TILE_74: load %arg0[%[[ii]], %[[jj]]]
|
|
load %arg0[%i, %j]: memref<?x?xf32>
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// COMMON-LABEL: @triangular
|
|
func @triangular(%arg0: memref<?x?xf32>) {
|
|
%c2 = constant 2 : index
|
|
%c44 = constant 44 : index
|
|
%c1 = constant 1 : index
|
|
// Range of the original outer loop:
|
|
// (upper - lower + step - 1) / step
|
|
// where step is known to be %c1.
|
|
// COMMON: %[[diff:.*]] = subi %c44, %c2
|
|
// COMMON: %[[adjustment:.*]] = subi %c1, %c1_{{.*}}
|
|
// COMMON-NEXT: %[[diff_adj:.*]] = addi %[[diff]], %[[adjustment]]
|
|
// COMMON-NEXT: %[[range:.*]] = divis %[[diff_adj]], %c1
|
|
|
|
// Ceildiv to get the parametric tile size.
|
|
// COMMON: %[[sum:.*]] = addi %[[range]], %c6
|
|
// COMMON-NEXT: %[[size:.*]] = divis %[[sum]], %c7
|
|
// New outer step (original is %c1).
|
|
// COMMON-NEXT: %[[step:.*]] = muli %c1, %[[size]]
|
|
|
|
// Constant adjustement for inner loop has been hoisted out.
|
|
// TILE_74: %[[adjustment2:.*]] = subi %c2, %c1_{{.*}}
|
|
|
|
// New outer loop.
|
|
// COMMON: loop.for %[[i:.*]] = %c2 to %c44 step %[[step]]
|
|
|
|
// Range of the original inner loop
|
|
// (upper - lower + step - 1) / step
|
|
// where step is known to be %c2.
|
|
// TILE_74: %[[diff2:.*]] = subi %[[i]], %c1
|
|
// TILE_74-NEXT: %[[diff2_adj:.*]] = addi %[[diff2]], %[[adjustment2]]
|
|
// TILE_74-NEXT: %[[range2:.*]] = divis %[[diff2_adj]], %c2
|
|
|
|
// Ceildiv to get the parametric tile size for the second original loop.
|
|
// TILE_74: %[[sum2:.*]] = addi %[[range2]], %c3
|
|
// TILE_74-NEXT: %[[size2:.*]] = divis %[[sum2]], %c4
|
|
// New inner step (original is %c2).
|
|
// TILE_74-NEXT: %[[step2:.*]] = muli %c2, %[[size2]]
|
|
|
|
// New inner loop.
|
|
// TILE_74:loop.for %[[j:.*]] = %c1 to %[[i]] step %[[step2]]
|
|
loop.for %i = %c2 to %c44 step %c1 {
|
|
// Upper bound for the inner loop min(%i + %step, %c44).
|
|
// COMMON: %[[stepped:.*]] = addi %[[i]], %[[step]]
|
|
// COMMON-NEXT: cmpi "slt", %c44, %[[stepped]]
|
|
// COMMON-NEXT: %[[ub:.*]] = select {{.*}}, %c44, %[[stepped]]
|
|
// TILE_74: %[[stepped2:.*]] = addi %[[j]], %[[step2]]
|
|
// TILE_74-NEXT: cmpi "slt", %[[i]], %[[stepped2]]
|
|
// TILE_74-NEXT: %[[ub2:.*]] = select {{.*}}, %[[i]], %[[stepped2]]
|
|
//
|
|
// Created inner loop.
|
|
// COMMON:loop.for %[[ii:.*]] = %[[i]] to %[[ub:.*]] step %c1
|
|
|
|
// This loop is not modified in TILE_7 case.
|
|
// TILE_7: loop.for %[[j:.*]] = %c1 to %[[ii]] step %c2
|
|
//
|
|
// But is modified in TILE_74 case.
|
|
// TILE_74:loop.for %[[jj:.*]] = %[[j]] to %[[ub2]] step %c2
|
|
loop.for %j = %c1 to %i step %c2 {
|
|
// The right iterator are used.
|
|
// TILE_7: load %arg0[%[[ii]], %[[j]]]
|
|
// TILE_74: load %arg0[%[[ii]], %[[jj]]]
|
|
load %arg0[%i, %j]: memref<?x?xf32>
|
|
}
|
|
}
|
|
return
|
|
}
|