Lines Matching refs:tile

149 #tile = affine_map < (i)->(i floordiv 4, i mod 4) >
156 func.func @multiple_argument_type(%A: memref<16xf64, #tile>, %B: f64, %C: memref<8xf64, #tile>, %D:…
157   %a = affine.load %A[0] : memref<16xf64, #tile>
159   affine.store %p, %A[10] : memref<16xf64, #tile>
160   call @single_argument_type(%C): (memref<8xf64, #tile>) -> ()
173 func.func @single_argument_type(%C : memref<8xf64, #tile>) {
174   %a = memref.alloc(): memref<8xf64, #tile>
175   %b = memref.alloc(): memref<16xf64, #tile>
178   call @single_argument_type(%a): (memref<8xf64, #tile>) -> ()
179   call @single_argument_type(%C): (memref<8xf64, #tile>) -> ()
180 …call @multiple_argument_type(%b, %d, %a, %e): (memref<16xf64, #tile>, f64, memref<8xf64, #tile>, m…
195 func.func @non_memref_ret(%A: memref<8xf64, #tile>) -> i1 {
205 …_multiple_argument_type(%A: memref<16xf64, #tile>, %B: f64, %C: memref<8xf64, #tile>) -> (memref<8…
206   %a = affine.load %A[0] : memref<16xf64, #tile>
211 …l @ret_single_argument_type(%C) : (memref<8xf64, #tile>) -> (memref<16xf64, #tile>, memref<8xf64, …
212     return %res2, %p: memref<8xf64, #tile>, f64
214     return %C, %p: memref<8xf64, #tile>, f64
229 …func @ret_single_argument_type(%C: memref<8xf64, #tile>) -> (memref<16xf64, #tile>, memref<8xf64, …
230   %a = memref.alloc() : memref<8xf64, #tile>
231   %b = memref.alloc() : memref<16xf64, #tile>
233 …l @ret_single_argument_type(%a) : (memref<8xf64, #tile>) -> (memref<16xf64, #tile>, memref<8xf64, …
234 …l @ret_single_argument_type(%C) : (memref<8xf64, #tile>) -> (memref<16xf64, #tile>, memref<8xf64, …
235 …le_argument_type(%b, %d, %a) : (memref<16xf64, #tile>, f64, memref<8xf64, #tile>) -> (memref<8xf64…
236 …ret_single_argument_type(%res1) : (memref<8xf64, #tile>) -> (memref<16xf64, #tile>, memref<8xf64, …
237   return %b, %a: memref<16xf64, #tile>, memref<8xf64, #tile>
252 func.func @func_A(%A: memref<8xf64, #tile>) {
253   call @func_B(%A) : (memref<8xf64, #tile>) -> ()
260 func.func @func_B(%A: memref<8xf64, #tile>) {
261   call @func_C(%A) : (memref<8xf64, #tile>) -> ()
268 func.func @func_C(%A: memref<8xf64, #tile>) {
275 func.func @some_func_A(%A: memref<8xf64, #tile>) {
276   call @some_func_B(%A) : (memref<8xf64, #tile>) -> ()
283 func.func @some_func_B(%A: memref<8xf64, #tile>) {
284   "test.test"(%A) : (memref<8xf64, #tile>) -> ()
285   call @some_func_C(%A) : (memref<8xf64, #tile>) -> ()
292 func.func @some_func_C(%A: memref<8xf64, #tile>) {
299 func.func private @external_func_A(memref<16xf64, #tile>) -> ()
303 func.func private @external_func_B(memref<16xf64, #tile>, f64) -> (memref<8xf64, #tile>)
307   %a = memref.alloc() : memref<16xf64, #tile>
308   call @external_func_A(%a) : (memref<16xf64, #tile>) -> ()
316 func.func @use_value_of_external(%A: memref<16xf64, #tile>, %B: f64) -> (memref<8xf64, #tile>) {
317   %res = call @external_func_B(%A, %B) : (memref<16xf64, #tile>, f64) -> (memref<8xf64, #tile>)
318   return %res : memref<8xf64, #tile>
324 func.func @affine_parallel_norm() ->  memref<8xf32, #tile> {
326   %a = memref.alloc() : memref<8xf32, #tile>
328   %1 = affine.parallel (%i) = (0) to (8) reduce ("assign") ->  memref<8xf32, #tile> {
329     affine.store %c, %a[%i] : memref<8xf32, #tile>
331     affine.yield %a : memref<8xf32, #tile>
333   return %1 : memref<8xf32, #tile>