30 lines
1.3 KiB
MLIR
30 lines
1.3 KiB
MLIR
// RUN: onnx-mlir-opt --normalize-memrefs %s -split-input-file | FileCheck %s
|
|
|
|
#map_tile = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2 floordiv 32, d3 floordiv 32, d2 mod 32, d3 mod 32)>
|
|
|
|
// CHECK-LABEL: test_krnl_memcpy_norm
|
|
// CHECK-SAME: -> memref<1x16x1x1x32x32xf32> {
|
|
func @test_krnl_memcpy_norm(%arg0: memref<1x16384xf32>) -> memref<1x16x4x4xf32, #map_tile> {
|
|
%0 = alloc() : memref<1x16x4x4xf32, #map_tile>
|
|
// CHECK: [[ALLOC:%.+]] = alloc() : memref<1x16x1x1x32x32xf32>
|
|
%c16384 = constant 16384 : i64
|
|
"krnl.memcpy"(%0, %arg0, %c16384) : (memref<1x16x4x4xf32, #map_tile>, memref<1x16384xf32>, i64) -> ()
|
|
// CHECK: "krnl.memcpy"
|
|
// CHECK-SAME: : (memref<1x16x1x1x32x32xf32>, memref<1x16384xf32>
|
|
return %0 : memref<1x16x4x4xf32, #map_tile>
|
|
// CHECK: return [[ALLOC]] : memref<1x16x1x1x32x32xf32>
|
|
}
|
|
|
|
// CHECK-LABEL: test_getref_norm
|
|
func @test_getref_norm() -> () {
|
|
%c0_i64 = constant 0 : i64
|
|
%0 = alloc() : memref<1x81920xf32>
|
|
%1 = alloc() : memref<1x16x4x4xf32, #map_tile>
|
|
// CHECK: [[ALLOC:%.+]] = alloc() : memref<1x16x1x1x32x32xf32>
|
|
%2 = "krnl.getref"(%0, %c0_i64) : (memref<1x81920xf32>, i64) -> memref<1x16x4x4xf32>
|
|
// Do something using %1 and %2
|
|
dealloc %1: memref<1x16x4x4xf32, #map_tile>
|
|
// CHECK: dealloc [[ALLOC:%.+]] : memref<1x16x1x1x32x32xf32>
|
|
dealloc %0: memref<1x81920xf32>
|
|
return
|
|
} |