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
52
53
54
55
56
57
58
59
60
61
62
63
|
; RUN: opt -S -mtriple=aarch64-linux-gnu -aarch64-sme-abi %s | FileCheck %s
; RUN: opt -S -mtriple=aarch64-linux-gnu -aarch64-sme-abi -aarch64-sme-abi %s | FileCheck %s
declare void @shared_za_callee() "aarch64_pstate_za_shared"
define void @private_za() "aarch64_pstate_za_new" {
; CHECK-LABEL: @private_za(
; CHECK-NEXT: prelude:
; CHECK-NEXT: [[TPIDR2:%.*]] = call i64 @llvm.aarch64.sme.get.tpidr2()
; CHECK-NEXT: [[CMP:%.*]] = icmp ne i64 [[TPIDR2]], 0
; CHECK-NEXT: br i1 [[CMP]], label [[SAVE_ZA:%.*]], label [[TMP0:%.*]]
; CHECK: save.za:
; CHECK-NEXT: call aarch64_sme_preservemost_from_x0 void @__arm_tpidr2_save()
; CHECK-NEXT: call void @llvm.aarch64.sme.set.tpidr2(i64 0)
; CHECK-NEXT: br label [[TMP0]]
; CHECK: 0:
; CHECK-NEXT: call void @llvm.aarch64.sme.za.enable()
; CHECK-NEXT: call void @shared_za_callee()
; CHECK-NEXT: call void @llvm.aarch64.sme.za.disable()
; CHECK-NEXT: ret void
;
call void @shared_za_callee()
ret void
}
define i32 @private_za_multiple_exit(i32 %a, i32 %b, i64 %cond) "aarch64_pstate_za_new" {
; CHECK-LABEL: @private_za_multiple_exit(
; CHECK-NEXT: prelude:
; CHECK-NEXT: [[TPIDR2:%.*]] = call i64 @llvm.aarch64.sme.get.tpidr2()
; CHECK-NEXT: [[CMP:%.*]] = icmp ne i64 [[TPIDR2]], 0
; CHECK-NEXT: br i1 [[CMP]], label [[SAVE_ZA:%.*]], label [[ENTRY:%.*]]
; CHECK: save.za:
; CHECK-NEXT: call aarch64_sme_preservemost_from_x0 void @__arm_tpidr2_save()
; CHECK-NEXT: call void @llvm.aarch64.sme.set.tpidr2(i64 0)
; CHECK-NEXT: br label [[ENTRY]]
; CHECK: entry:
; CHECK-NEXT: call void @llvm.aarch64.sme.za.enable()
; CHECK-NEXT: [[TOBOOL:%.*]] = icmp eq i64 [[COND:%.*]], 1
; CHECK-NEXT: br i1 [[TOBOOL]], label [[IF_ELSE:%.*]], label [[IF_END:%.*]]
; CHECK: if.else:
; CHECK-NEXT: [[ADD:%.*]] = add i32 [[A:%.*]], [[B:%.*]]
; CHECK-NEXT: call void @llvm.aarch64.sme.za.disable()
; CHECK-NEXT: ret i32 [[ADD]]
; CHECK: if.end:
; CHECK-NEXT: [[SUB:%.*]] = sub i32 [[A]], [[B]]
; CHECK-NEXT: call void @llvm.aarch64.sme.za.disable()
; CHECK-NEXT: ret i32 [[SUB]]
;
entry:
%tobool = icmp eq i64 %cond, 1
br i1 %tobool, label %if.else, label %if.end
if.else:
%add = add i32 %a, %b
ret i32 %add
if.end:
%sub = sub i32 %a, %b
ret i32 %sub
}
; CHECK: declare void @__arm_tpidr2_save() #[[ATTR:[0-9]+]]
; CHECK: attributes #[[ATTR]] = { "aarch64_pstate_sm_compatible" "aarch64_pstate_za_preserved" }
|
