a14a2805b0
When dealing with dialects that will results in function calls to external libraries, it is important to be able to handle maps as some dialects may require mapped data. Before this patch, the detection of whether normalization can apply or not, operations are compared to an explicit list of operations (`alloc`, `dealloc`, `return`) or to the presence of specific operation interfaces (`AffineReadOpInterface`, `AffineWriteOpInterface`, `AffineDMAStartOp`, or `AffineDMAWaitOp`). This patch add a trait, `MemRefsNormalizable` to determine if an operation can have its `memrefs` normalized. This trait can be used in turn by dialects to assert that such operations are compatible with normalization of `memrefs` with nontrivial memory layout specification. An example is given in the literal tests. Differential Revision: https://reviews.llvm.org/D86236
58 lines
2.7 KiB
MLIR
58 lines
2.7 KiB
MLIR
// RUN: mlir-opt -normalize-memrefs %s | FileCheck %s
|
|
|
|
// For all these cases, we test if MemRefs Normalization works with the test
|
|
// operations.
|
|
// * test.op_norm: this operation has the MemRefsNormalizable attribute. The tests
|
|
// that include this operation are constructed so that the normalization should
|
|
// happen.
|
|
// * test_op_nonnorm: this operation does not have the MemRefsNormalization
|
|
// attribute. The tests that include this operation are contructed so that the
|
|
// normalization should not happen.
|
|
|
|
#map0 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2 floordiv 32, d3 floordiv 64, d2 mod 32, d3 mod 64)>
|
|
|
|
// Test with op_norm and maps in arguments and in the operations in the function.
|
|
|
|
// CHECK-LABEL: test_norm
|
|
// CHECK-SAME: (%[[ARG0:[a-z0-9]*]]: memref<1x16x1x1x32x64xf32>)
|
|
func @test_norm(%arg0 : memref<1x16x14x14xf32, #map0>) -> () {
|
|
%0 = alloc() : memref<1x16x14x14xf32, #map0>
|
|
"test.op_norm"(%arg0, %0) : (memref<1x16x14x14xf32, #map0>, memref<1x16x14x14xf32, #map0>) -> ()
|
|
dealloc %0 : memref<1x16x14x14xf32, #map0>
|
|
|
|
// CHECK: %[[v0:[a-z0-9]*]] = alloc() : memref<1x16x1x1x32x64xf32>
|
|
// CHECK: "test.op_norm"(%[[ARG0]], %[[v0]]) : (memref<1x16x1x1x32x64xf32>, memref<1x16x1x1x32x64xf32>) -> ()
|
|
// CHECK: dealloc %[[v0]] : memref<1x16x1x1x32x64xf32>
|
|
return
|
|
}
|
|
|
|
// Same test with op_nonnorm, with maps in the argmentets and the operations in the function.
|
|
|
|
// CHECK-LABEL: test_nonnorm
|
|
// CHECK-SAME: (%[[ARG0:[a-z0-9]*]]: memref<1x16x14x14xf32, #map0>)
|
|
func @test_nonnorm(%arg0 : memref<1x16x14x14xf32, #map0>) -> () {
|
|
%0 = alloc() : memref<1x16x14x14xf32, #map0>
|
|
"test.op_nonnorm"(%arg0, %0) : (memref<1x16x14x14xf32, #map0>, memref<1x16x14x14xf32, #map0>) -> ()
|
|
dealloc %0 : memref<1x16x14x14xf32, #map0>
|
|
|
|
// CHECK: %[[v0:[a-z0-9]*]] = alloc() : memref<1x16x14x14xf32, #map0>
|
|
// CHECK: "test.op_nonnorm"(%[[ARG0]], %[[v0]]) : (memref<1x16x14x14xf32, #map0>, memref<1x16x14x14xf32, #map0>) -> ()
|
|
// CHECK: dealloc %[[v0]] : memref<1x16x14x14xf32, #map0>
|
|
return
|
|
}
|
|
|
|
// Test with op_norm, with maps in the operations in the function.
|
|
|
|
// CHECK-LABEL: test_norm_mix
|
|
// CHECK-SAME: (%[[ARG0:[a-z0-9]*]]: memref<1x16x1x1x32x64xf32>
|
|
func @test_norm_mix(%arg0 : memref<1x16x1x1x32x64xf32>) -> () {
|
|
%0 = alloc() : memref<1x16x14x14xf32, #map0>
|
|
"test.op_norm"(%arg0, %0) : (memref<1x16x1x1x32x64xf32>, memref<1x16x14x14xf32, #map0>) -> ()
|
|
dealloc %0 : memref<1x16x14x14xf32, #map0>
|
|
|
|
// CHECK: %[[v0:[a-z0-9]*]] = alloc() : memref<1x16x1x1x32x64xf32>
|
|
// CHECK: "test.op_norm"(%[[ARG0]], %[[v0]]) : (memref<1x16x1x1x32x64xf32>, memref<1x16x1x1x32x64xf32>) -> ()
|
|
// CHECK: dealloc %[[v0]] : memref<1x16x1x1x32x64xf32>
|
|
return
|
|
}
|