Lines Matching refs:memref
15 func.func @condBranch(%arg0: i1, %arg1: memref<2xf32>, %arg2: memref<2xf32>) {
18 cf.br ^bb3(%arg1 : memref<2xf32>)
20 %0 = memref.alloc() : memref<2xf32>
21 test.buffer_based in(%arg1: memref<2xf32>) out(%0: memref<2xf32>)
22 cf.br ^bb3(%0 : memref<2xf32>)
23 ^bb3(%1: memref<2xf32>):
24 test.copy(%1, %arg2) : (memref<2xf32>, memref<2xf32>)
29 // CHECK: %[[ALLOC:.*]] = memref.alloc()
46 %arg1: memref<?xf32>,
47 %arg2: memref<?xf32>,
51 cf.br ^bb3(%arg1 : memref<?xf32>)
53 %1 = memref.alloc(%0) : memref<?xf32>
54 test.buffer_based in(%arg1: memref<?xf32>) out(%1: memref<?xf32>)
55 cf.br ^bb3(%1 : memref<?xf32>)
56 ^bb3(%2: memref<?xf32>):
57 test.copy(%2, %arg2) : (memref<?xf32>, memref<?xf32>)
64 // CHECK-NEXT: %[[ALLOC0:.*]] = memref.alloc(%[[IDX]])
78 %arg1: memref<2xf32>,
79 %arg2: memref<2xf32>) {
82 cf.br ^bb3(%arg1 : memref<2xf32>)
84 %0 = memref.alloc() : memref<2xf32>
85 test.region_buffer_based in(%arg1: memref<2xf32>) out(%0: memref<2xf32>) {
87 %1 = memref.alloc() : memref<2xf32>
88 test.buffer_based in(%arg1: memref<2xf32>) out(%1: memref<2xf32>)
92 cf.br ^bb3(%0 : memref<2xf32>)
93 ^bb3(%1: memref<2xf32>):
94 test.copy(%1, %arg2) : (memref<2xf32>, memref<2xf32>)
98 // CHECK: %[[ALLOC0:.*]] = memref.alloc()
100 // CHECK: %[[ALLOC1:.*]] = memref.alloc()
112 %arg1 : index) -> memref<?x?xf32> {
114 %1 = memref.alloc(%arg0, %arg0) : memref<?x?xf32>
115 %2 = scf.if %0 -> (memref<?x?xf32>) {
116 scf.yield %1 : memref<?x?xf32>
118 %3 = memref.alloc(%arg0, %arg1) : memref<?x?xf32>
119 scf.yield %1 : memref<?x?xf32>
121 return %2 : memref<?x?xf32>
124 // CHECK: %[[ALLOC0:.*]] = memref.alloc(%arg0, %arg0)
127 // CHECK-NEXT: %[[ALLOC1:.*]] = memref.alloc(%arg0, %arg1)
139 %buf: memref<2xf32>,
140 %res: memref<2xf32>) {
141 %0 = memref.alloc() : memref<2xf32>
143 iter_args(%iterBuf = %buf) -> memref<2xf32> {
145 %3 = memref.alloc() : memref<2xf32>
146 scf.yield %3 : memref<2xf32>
148 test.copy(%1, %res) : (memref<2xf32>, memref<2xf32>)
152 // CHECK: %[[ALLOC0:.*]] = memref.alloc()
154 // CHECK: %[[ALLOC1:.*]] = memref.alloc()
167 %buf: memref<2xf32>) -> memref<2xf32> {
168 %0 = memref.alloc() : memref<2xf32>
170 iter_args(%iterBuf = %buf) -> memref<2xf32> {
172 %3 = scf.if %2 -> (memref<2xf32>) {
173 %4 = memref.alloc() : memref<2xf32>
174 scf.yield %4 : memref<2xf32>
176 scf.yield %0 : memref<2xf32>
178 scf.yield %3 : memref<2xf32>
180 return %1 : memref<2xf32>
183 // CHECK: %[[ALLOC0:.*]] = memref.alloc()
185 // CHECK: %[[ALLOC1:.*]] = memref.alloc()
199 %buf: memref<2xf32>,
200 %res: memref<2xf32>) {
201 %0 = memref.alloc() : memref<2xf32>
203 iter_args(%iterBuf = %buf) -> memref<2xf32> {
205 iter_args(%iterBuf2 = %iterBuf) -> memref<2xf32> {
207 iter_args(%iterBuf3 = %iterBuf2) -> memref<2xf32> {
208 %4 = memref.alloc() : memref<2xf32>
210 %6 = scf.if %5 -> (memref<2xf32>) {
211 %7 = memref.alloc() : memref<2xf32>
212 %8 = memref.alloc() : memref<2xf32>
213 scf.yield %8 : memref<2xf32>
215 scf.yield %iterBuf3 : memref<2xf32>
217 %9 = memref.alloc() : memref<2xf32>
218 scf.yield %6 : memref<2xf32>
220 scf.yield %3 : memref<2xf32>
222 scf.yield %2 : memref<2xf32>
224 test.copy(%1, %res) : (memref<2xf32>, memref<2xf32>)
228 // CHECK: %[[ALLOC0:.*]] = memref.alloc()
229 // CHECK-NEXT: %[[ALLOC1:.*]] = memref.alloc()
230 // CHECK-NEXT: %[[ALLOC2:.*]] = memref.alloc()
235 // CHECK: %[[ALLOC3:.*]] = memref.alloc()
236 // CHECK: %[[ALLOC4:.*]] = memref.alloc()
246 %buf: memref<?xf32>,
247 %res: memref<?xf32>) {
248 %0 = memref.alloc(%arg0) : memref<?xf32>
250 iter_args(%iterBuf = %buf) -> memref<?xf32> {
252 iter_args(%iterBuf2 = %iterBuf) -> memref<?xf32> {
254 iter_args(%iterBuf3 = %iterBuf2) -> memref<?xf32> {
255 %4 = memref.alloc(%i3) : memref<?xf32>
257 %6 = scf.if %5 -> (memref<?xf32>) {
258 %7 = memref.alloc(%i3) : memref<?xf32>
259 scf.yield %7 : memref<?xf32>
261 scf.yield %iterBuf3 : memref<?xf32>
263 %8 = memref.alloc(%i3) : memref<?xf32>
264 scf.yield %6 : memref<?xf32>
266 scf.yield %3 : memref<?xf32>
268 scf.yield %0 : memref<?xf32>
270 test.copy(%1, %res) : (memref<?xf32>, memref<?xf32>)
274 // CHECK: %[[ALLOC0:.*]] = memref.alloc({{.*}})
278 // CHECK: %[[ALLOC1:.*]] = memref.alloc({{.*}})
279 // CHECK: %[[ALLOC2:.*]] = memref.alloc({{.*}})
288 %buf: memref<2xf32>,
289 %res: memref<2xf32>) {
290 %0 = memref.alloc() : memref<2xf32>
292 iter_args(%iterBuf = %buf) -> memref<2xf32> {
293 %2 = memref.alloc() : memref<2xf32>
294 scf.yield %0 : memref<2xf32>
296 test.copy(%1, %res) : (memref<2xf32>, memref<2xf32>)
300 // CHECK: %[[ALLOC0:.*]] = memref.alloc({{.*}})
301 // CHECK-NEXT: %[[ALLOC1:.*]] = memref.alloc({{.*}})
311 %buf: memref<2xf32>,
312 %res: memref<2xf32>) {
314 iter_args(%iterBuf = %buf) -> memref<2xf32> {
315 %1 = memref.alloc() : memref<2xf32>
316 scf.yield %1 : memref<2xf32>
318 test.copy(%0, %res) : (memref<2xf32>, memref<2xf32>)
323 // CHECK-NEXT: %[[ALLOC0:.*]] = memref.alloc({{.*}})
332 %buf: memref<2xf32>,
333 %res: memref<2xf32>) {
334 %0 = memref.alloc() : memref<2xf32>
336 iter_args(%iterBuf = %buf) -> memref<2xf32> {
338 iter_args(%iterBuf2 = %iterBuf) -> memref<2xf32> {
339 %3 = memref.alloc() : memref<2xf32>
340 scf.yield %0 : memref<2xf32>
342 scf.yield %0 : memref<2xf32>
344 test.copy(%1, %res) : (memref<2xf32>, memref<2xf32>)
348 // CHECK: %[[ALLOC0:.*]] = memref.alloc({{.*}})
349 // CHECK-NEXT: %[[ALLOC1:.*]] = memref.alloc({{.*}})
359 %buf: memref<2xf32>,
360 %res: memref<2xf32>) {
362 iter_args(%iterBuf = %buf) -> memref<2xf32> {
364 %2 = scf.if %1 -> (memref<2xf32>) {
365 %3 = memref.alloc() : memref<2xf32>
366 scf.yield %3 : memref<2xf32>
368 scf.yield %iterBuf : memref<2xf32>
370 scf.yield %2 : memref<2xf32>
372 test.copy(%0, %res) : (memref<2xf32>, memref<2xf32>)
378 // CHECK-NEXT: %[[ALLOC0:.*]] = memref.alloc({{.*}})
387 %buf: memref<2xf32>,
388 %res: memref<2xf32>) {
389 %0 = memref.alloc() : memref<2xf32>
391 %2 = scf.if %1 -> (memref<2xf32>) {
393 iter_args(%iterBuf = %buf) -> memref<2xf32> {
394 %4 = memref.alloc() : memref<2xf32>
395 scf.yield %0 : memref<2xf32>
397 scf.yield %0 : memref<2xf32>
401 scf.yield %0 : memref<2xf32>
403 test.copy(%2, %res) : (memref<2xf32>, memref<2xf32>)
408 // CHECK-NEXT: %[[ALLOC0:.*]] = memref.alloc({{.*}})
418 %buf: memref<2xf32>,
419 %res: memref<2xf32>) {
420 %0 = memref.alloc() : memref<2xf32>
422 iter_args(%iterBuf = %buf) -> memref<2xf32> {
423 %2 = memref.alloc(%i) : memref<?xf32>
424 scf.yield %0 : memref<2xf32>
426 test.copy(%1, %res) : (memref<2xf32>, memref<2xf32>)
430 // CHECK: %[[ALLOC0:.*]] = memref.alloc({{.*}})
432 // CHECK-NEXT: %[[ALLOC1:.*]] = memref.alloc({{.*}})
441 %buf: memref<2xf32>,
442 %res: memref<2xf32>) {
443 %0 = memref.alloc() : memref<2xf32>
445 iter_args(%iterBuf = %buf) -> memref<2xf32> {
447 iter_args(%iterBuf2 = %iterBuf) -> memref<2xf32> {
448 %3 = memref.alloc(%i) : memref<?xf32>
449 scf.yield %0 : memref<2xf32>
451 scf.yield %0 : memref<2xf32>
453 test.copy(%1, %res) : (memref<2xf32>, memref<2xf32>)
457 // CHECK: %[[ALLOC0:.*]] = memref.alloc({{.*}})
459 // CHECK-NEXT: %[[ALLOC1:.*]] = memref.alloc({{.*}})
471 %buf: memref<2xf32>,
472 %res: memref<2xf32>) {
473 %0 = memref.alloca() : memref<2xf32>
475 iter_args(%iterBuf = %buf) -> memref<2xf32> {
477 iter_args(%iterBuf2 = %iterBuf) -> memref<2xf32> {
478 %3 = memref.alloca() : memref<2xf32>
479 scf.yield %0 : memref<2xf32>
481 scf.yield %0 : memref<2xf32>
483 test.copy(%1, %res) : (memref<2xf32>, memref<2xf32>)
487 // CHECK: %[[ALLOCA0:.*]] = memref.alloca({{.*}})
488 // CHECK-NEXT: %[[ALLOCA1:.*]] = memref.alloca({{.*}})