89bc449cee
Change the AsmPrinter to number values breadth-first so that values in adjacent regions can have the same name. This allows for ModuleOp to contain operations that produce results. This also standardizes the special name of region entry arguments to "arg[0-9+]" now that Functions are also operations. PiperOrigin-RevId: 257225069
38 lines
2.2 KiB
MLIR
38 lines
2.2 KiB
MLIR
// RUN: mlir-opt %s -affine-materialize-vectors -vector-size=4 -vector-size=4 | FileCheck %s
|
|
|
|
// CHECK-DAG: #[[ID1:map[0-9]+]] = (d0) -> (d0)
|
|
// CHECK-DAG: #[[D0D1D2D3TOD1D0:map[0-9]+]] = (d0, d1, d2, d3) -> (d1, d0)
|
|
// CHECK-DAG: #[[D0P1:map[0-9]+]] = (d0) -> (d0 + 1)
|
|
// CHECK-DAG: #[[D0P2:map[0-9]+]] = (d0) -> (d0 + 2)
|
|
// CHECK-DAG: #[[D0P3:map[0-9]+]] = (d0) -> (d0 + 3)
|
|
|
|
// CHECK-LABEL: func @materialize
|
|
func @materialize(%M : index, %N : index, %O : index, %P : index) {
|
|
%A = alloc (%M, %N, %O, %P) : memref<?x?x?x?xf32, 0>
|
|
%f1 = constant dense<1.000000e+00> : vector<4x4x4xf32>
|
|
// CHECK: affine.for %{{.*}} = 0 to %{{.*}} step 4 {
|
|
// CHECK-NEXT: affine.for %{{.*}} = 0 to %{{.*}} step 4 {
|
|
// CHECK-NEXT: affine.for %{{.*}} = 0 to %{{.*}} {
|
|
// CHECK-NEXT: affine.for %{{.*}} = 0 to %{{.*}} step 4 {
|
|
// CHECK-NEXT: %[[a:[0-9]+]] = {{.*}}[[ID1]](%{{.*}})
|
|
// CHECK-NEXT: %[[b:[0-9]+]] = {{.*}}[[ID1]](%{{.*}})
|
|
// CHECK-NEXT: %[[c:[0-9]+]] = {{.*}}[[ID1]](%{{.*}})
|
|
// CHECK-NEXT: %[[d:[0-9]+]] = {{.*}}[[ID1]](%{{.*}})
|
|
// CHECK-NEXT: vector.transfer_write {{.*}}, %{{.*}}[%[[a]], %[[b]], %[[c]], %[[d]]] {permutation_map = #[[D0D1D2D3TOD1D0]]} : vector<4x4xf32>, memref<?x?x?x?xf32>
|
|
// CHECK: %[[b1:[0-9]+]] = {{.*}}[[D0P1]](%{{.*}})
|
|
// CHECK: vector.transfer_write {{.*}}, %{{.*}}[{{.*}}, %[[b1]], {{.*}}] {permutation_map = #[[D0D1D2D3TOD1D0]]} : vector<4x4xf32>, memref<?x?x?x?xf32>
|
|
// CHECK: %[[b2:[0-9]+]] = {{.*}}[[D0P2]](%{{.*}})
|
|
// CHECK: vector.transfer_write {{.*}}, %{{.*}}[{{.*}}, %[[b2]], {{.*}}] {permutation_map = #[[D0D1D2D3TOD1D0]]} : vector<4x4xf32>, memref<?x?x?x?xf32>
|
|
// CHECK: %[[b3:[0-9]+]] = {{.*}}[[D0P3]](%{{.*}})
|
|
// CHECK: vector.transfer_write {{.*}}, %{{.*}}[{{.*}}, %[[b3]], {{.*}}] {permutation_map = #[[D0D1D2D3TOD1D0]]} : vector<4x4xf32>, memref<?x?x?x?xf32>
|
|
affine.for %i0 = 0 to %M step 4 {
|
|
affine.for %i1 = 0 to %N step 4 {
|
|
affine.for %i2 = 0 to %O {
|
|
affine.for %i3 = 0 to %P step 4 {
|
|
vector.transfer_write %f1, %A[%i0, %i1, %i2, %i3] {permutation_map = (d0, d1, d2, d3) -> (d3, d1, d0)} : vector<4x4x4xf32>, memref<?x?x?x?xf32, 0>
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return
|
|
} |