Lines Matching refs:memref

17 // CHECK-SAME: (%[[ARG0:.*]]: memref<1x16x1x1x32x64xf32>)
18 func.func @test_norm(%arg0 : memref<1x16x14x14xf32, #map0>) -> () {
19     %0 = memref.alloc() : memref<1x16x14x14xf32, #map0>
20     "test.op_norm"(%arg0, %0) : (memref<1x16x14x14xf32, #map0>, memref<1x16x14x14xf32, #map0>) -> ()
21     memref.dealloc %0 :  memref<1x16x14x14xf32, #map0>
23     // CHECK: %[[v0:.*]] = memref.alloc() : memref<1x16x1x1x32x64xf32>
24 …// CHECK: "test.op_norm"(%[[ARG0]], %[[v0]]) : (memref<1x16x1x1x32x64xf32>, memref<1x16x1x1x32x64x…
25     // CHECK: memref.dealloc %[[v0]] : memref<1x16x1x1x32x64xf32>
32 // CHECK-SAME: (%[[ARG0:.*]]: memref<1x16x14x14xf32, #map>)
33 func.func @test_nonnorm(%arg0 : memref<1x16x14x14xf32, #map0>) -> () {
34     %0 = memref.alloc() : memref<1x16x14x14xf32, #map0>
35 …"test.op_nonnorm"(%arg0, %0) : (memref<1x16x14x14xf32, #map0>, memref<1x16x14x14xf32, #map0>) -> ()
36     memref.dealloc %0 :  memref<1x16x14x14xf32, #map0>
38     // CHECK: %[[v0:.*]] = memref.alloc() : memref<1x16x14x14xf32, #map>
39 …// CHECK: "test.op_nonnorm"(%[[ARG0]], %[[v0]]) : (memref<1x16x14x14xf32, #map>, memref<1x16x14x14…
40     // CHECK: memref.dealloc %[[v0]] : memref<1x16x14x14xf32, #map>
47 // CHECK-SAME: (%[[ARG0:.*]]: memref<1x16x1x1x32x64xf32>
48 func.func @test_norm_mix(%arg0 : memref<1x16x1x1x32x64xf32>) -> () {
49     %0 = memref.alloc() : memref<1x16x14x14xf32, #map0>
50     "test.op_norm"(%arg0, %0) : (memref<1x16x1x1x32x64xf32>, memref<1x16x14x14xf32, #map0>) -> ()
51     memref.dealloc %0 :  memref<1x16x14x14xf32, #map0>
53     // CHECK: %[[v0:.*]] = memref.alloc() : memref<1x16x1x1x32x64xf32>
54 …// CHECK: "test.op_norm"(%[[ARG0]], %[[v0]]) : (memref<1x16x1x1x32x64xf32>, memref<1x16x1x1x32x64x…
55     // CHECK: memref.dealloc %[[v0]] : memref<1x16x1x1x32x64xf32>
64 // CHECK-SAME: (%[[ARG0:.*]]: memref<1x16x14x14xf32>
65 func.func @test_load_store(%arg0 : memref<1x16x14x14xf32>) -> () {
66     %0 = memref.alloc() : memref<1x16x14x14xf32, #map_tile>
67     // CHECK: %[[v0:.*]] = memref.alloc() : memref<1x16x1x1x32x32xf32>
68     %1 = memref.alloc() : memref<1x16x14x14xf32>
69     // CHECK: %[[v1:.*]] = memref.alloc() : memref<1x16x14x14xf32>
70     "test.op_norm"(%0, %1) : (memref<1x16x14x14xf32, #map_tile>, memref<1x16x14x14xf32>) -> ()
71 …// CHECK: "test.op_norm"(%[[v0]], %[[v1]]) : (memref<1x16x1x1x32x32xf32>, memref<1x16x14x14xf32>) …
77             %2 = memref.load %1[%i, %j, %k, %l] : memref<1x16x14x14xf32>
78             // CHECK: memref<1x16x14x14xf32>
80             memref.store %3, %arg0[%i, %j, %k, %l] : memref<1x16x14x14xf32>
81             // CHECK: memref<1x16x14x14xf32>
86     memref.dealloc %0 :  memref<1x16x14x14xf32, #map_tile>
87     // CHECK: memref.dealloc %[[v0]] : memref<1x16x1x1x32x32xf32>
88     memref.dealloc %1 :  memref<1x16x14x14xf32>
89     // CHECK: memref.dealloc %[[v1]] : memref<1x16x14x14xf32>
96 // CHECK-SAME: (%[[ARG0:.*]]: memref<1x16x1x1x32x32xf32>) -> (memref<1x16x1x1x32x32xf32>, memref<1x…
97 func.func @test_norm_ret(%arg0: memref<1x16x14x14xf32, #map_tile>) -> (memref<1x16x14x14xf32, #map_…
98     %0 = memref.alloc() : memref<1x16x14x14xf32, #map_tile>
99     // CHECK-NEXT: %[[v0:.*]] = memref.alloc() : memref<1x16x1x1x32x32xf32>
100 …, %2 = "test.op_norm_ret"(%arg0) : (memref<1x16x14x14xf32, #map_tile>) -> (memref<1x16x14x14xf32, …
102 …// CHECK-SAME: (memref<1x16x1x1x32x32xf32>) -> (memref<1x16x1x1x32x32xf32>, memref<1x16x14x14xf32>)
103 …"test.op_norm"(%1, %0) : (memref<1x16x14x14xf32, #map_tile>, memref<1x16x14x14xf32, #map_tile>) ->…
105     // CHECK-SAME: : (memref<1x16x1x1x32x32xf32>, memref<1x16x1x1x32x32xf32>) -> ()
106     memref.dealloc %0 : memref<1x16x14x14xf32, #map_tile>
107     // CHECK-NEXT: memref.dealloc %[[v0]] : memref<1x16x1x1x32x32xf32>
108     return %1, %2 : memref<1x16x14x14xf32, #map_tile>, memref<1x16x14x14xf32>
109     // CHECK-NEXT: return %[[v1]], %[[v2]] : memref<1x16x1x1x32x32xf32>, memref<1x16x14x14xf32>
121 // Test with memref.reinterpret_cast
124 // CHECK-SAME: (%[[ARG0:.*]]: memref<1x32xf32>) -> memref<3x1x1xf32> {
125 func.func @test_norm_reinterpret_cast(%arg0 : memref<3xf32, #map_1d_tile>) -> (memref<3x1x1xf32>) {
126     %0 = memref.alloc() : memref<3xf32>
127     "test.op_norm"(%arg0, %0) : (memref<3xf32, #map_1d_tile>, memref<3xf32>) -> ()
128 …%1 = memref.reinterpret_cast %0 to offset: [0], sizes: [3, 1, 1], strides: [1, 1, 1] : memref<3xf3…
129     // CHECK: %[[v0:.*]] = memref.alloc() : memref<3xf32>
130     // CHECK: "test.op_norm"(%[[ARG0]], %[[v0]]) : (memref<1x32xf32>, memref<3xf32>) -> ()
131 …// CHECK: memref.reinterpret_cast %[[v0]] to offset: [0], sizes: [3, 1, 1], strides: [1, 1, 1] : m…
132     return %1 : memref<3x1x1xf32>