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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
// RUN: mlir-opt -finalize-memref-to-llvm %s | FileCheck %s
// CHECK-LABEL: @empty
func.func @empty() {
// CHECK: llvm.intr.stacksave
// CHECK: llvm.br
memref.alloca_scope {
memref.alloca_scope.return
}
// CHECK: llvm.intr.stackrestore
// CHECK: llvm.br
return
}
// CHECK-LABEL: @returns_nothing
func.func @returns_nothing(%b: f32) {
%a = arith.constant 10.0 : f32
// CHECK: llvm.intr.stacksave
memref.alloca_scope {
%c = arith.addf %a, %b : f32
memref.alloca_scope.return
}
// CHECK: llvm.intr.stackrestore
return
}
// CHECK-LABEL: @returns_one_value
func.func @returns_one_value(%b: f32) -> f32 {
%a = arith.constant 10.0 : f32
// CHECK: llvm.intr.stacksave
%result = memref.alloca_scope -> f32 {
%c = arith.addf %a, %b : f32
memref.alloca_scope.return %c: f32
}
// CHECK: llvm.intr.stackrestore
return %result : f32
}
// CHECK-LABEL: @returns_multiple_values
func.func @returns_multiple_values(%b: f32) -> f32 {
%a = arith.constant 10.0 : f32
// CHECK: llvm.intr.stacksave
%result1, %result2 = memref.alloca_scope -> (f32, f32) {
%c = arith.addf %a, %b : f32
%d = arith.subf %a, %b : f32
memref.alloca_scope.return %c, %d: f32, f32
}
// CHECK: llvm.intr.stackrestore
%result = arith.addf %result1, %result2 : f32
return %result : f32
}
|
