1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
// RUN: mlir-opt %s --pass-pipeline='builtin.module(func.func(sroa))' --split-input-file | FileCheck %s
// Verifies that allocators with mutliple slots are handled properly.
// CHECK-LABEL: func.func @multi_slot_alloca
func.func @multi_slot_alloca() -> (i32, i32) {
%0 = arith.constant 0 : index
%1, %2 = test.multi_slot_alloca : () -> (memref<2xi32>, memref<4xi32>)
// CHECK-COUNT-2: test.multi_slot_alloca : () -> memref<i32>
%3 = memref.load %1[%0] {first}: memref<2xi32>
%4 = memref.load %2[%0] {second} : memref<4xi32>
return %3, %4 : i32, i32
}
// -----
// Verifies that a multi slot allocator can be partially destructured.
func.func private @consumer(memref<2xi32>)
// CHECK-LABEL: func.func @multi_slot_alloca_only_second
func.func @multi_slot_alloca_only_second() -> (i32, i32) {
%0 = arith.constant 0 : index
// CHECK: test.multi_slot_alloca : () -> memref<2xi32>
// CHECK: test.multi_slot_alloca : () -> memref<i32>
%1, %2 = test.multi_slot_alloca : () -> (memref<2xi32>, memref<4xi32>)
func.call @consumer(%1) : (memref<2xi32>) -> ()
%3 = memref.load %1[%0] : memref<2xi32>
%4 = memref.load %2[%0] : memref<4xi32>
return %3, %4 : i32, i32
}
|
