aboutsummaryrefslogtreecommitdiff
path: root/llvm/test/CodeGen/ARM/bf16-imm.ll
blob: 7532bbcd09b3ff2cb66f69c680d2d10e3a1c9541 (plain)
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -mtriple=armv8.6a-none-none-eabi < %s | FileCheck %s --check-prefixes=CHECK
; RUN: llc -mtriple=armv8.6a-none-none-eabi -mattr=+bf16,+neon < %s | FileCheck %s --check-prefixes=CHECK
; RUN: llc -mtriple=armv8.6a-none-none-eabi -mattr=+bf16,+neon,+fullfp16 < %s | FileCheck %s --check-prefix=CHECK-FP16
; RUN: llc -mtriple=armv8.6a-none-none-eabi -mattr=+bf16,+neon -float-abi=hard < %s | FileCheck %s --check-prefix=CHECK-HARD

define bfloat @bitcast_zero() {
; CHECK-LABEL: bitcast_zero:
; CHECK:       @ %bb.0:
; CHECK-NEXT:    .pad #4
; CHECK-NEXT:    sub sp, sp, #4
; CHECK-NEXT:    mov r0, #0
; CHECK-NEXT:    strh r0, [sp, #2]
; CHECK-NEXT:    ldrh r0, [sp, #2]
; CHECK-NEXT:    add sp, sp, #4
; CHECK-NEXT:    bx lr
;
; CHECK-FP16-LABEL: bitcast_zero:
; CHECK-FP16:       @ %bb.0:
; CHECK-FP16-NEXT:    mov r0, #0
; CHECK-FP16-NEXT:    vmov.f16 s0, r0
; CHECK-FP16-NEXT:    vmov r0, s0
; CHECK-FP16-NEXT:    bx lr
;
; CHECK-HARD-LABEL: bitcast_zero:
; CHECK-HARD:       @ %bb.0:
; CHECK-HARD-NEXT:    .pad #4
; CHECK-HARD-NEXT:    sub sp, sp, #4
; CHECK-HARD-NEXT:    mov r0, #0
; CHECK-HARD-NEXT:    strh r0, [sp, #2]
; CHECK-HARD-NEXT:    ldrh r0, [sp, #2]
; CHECK-HARD-NEXT:    vmov s0, r0
; CHECK-HARD-NEXT:    add sp, sp, #4
; CHECK-HARD-NEXT:    bx lr
  %z = bitcast i16 0 to bfloat
  ret bfloat %z
}

define bfloat @zero() {
; CHECK-LABEL: zero:
; CHECK:       @ %bb.0:
; CHECK-NEXT:    adr r0, .LCPI1_0
; CHECK-NEXT:    ldrh r0, [r0]
; CHECK-NEXT:    bx lr
; CHECK-NEXT:    .p2align 1
; CHECK-NEXT:  @ %bb.1:
; CHECK-NEXT:  .LCPI1_0:
; CHECK-NEXT:    .short 0x0000 @ bfloat 0
;
; CHECK-FP16-LABEL: zero:
; CHECK-FP16:       @ %bb.0:
; CHECK-FP16-NEXT:    vldr.16 s0, .LCPI1_0
; CHECK-FP16-NEXT:    vmov r0, s0
; CHECK-FP16-NEXT:    bx lr
; CHECK-FP16-NEXT:    .p2align 1
; CHECK-FP16-NEXT:  @ %bb.1:
; CHECK-FP16-NEXT:  .LCPI1_0:
; CHECK-FP16-NEXT:    .short 0x0000 @ bfloat 0
;
; CHECK-HARD-LABEL: zero:
; CHECK-HARD:       @ %bb.0:
; CHECK-HARD-NEXT:    adr r0, .LCPI1_0
; CHECK-HARD-NEXT:    ldrh r0, [r0]
; CHECK-HARD-NEXT:    vmov s0, r0
; CHECK-HARD-NEXT:    bx lr
; CHECK-HARD-NEXT:    .p2align 1
; CHECK-HARD-NEXT:  @ %bb.1:
; CHECK-HARD-NEXT:  .LCPI1_0:
; CHECK-HARD-NEXT:    .short 0x0000 @ bfloat 0
  ret bfloat 0xR0000
}

define bfloat @bitcast_tenk() {
; CHECK-LABEL: bitcast_tenk:
; CHECK:       @ %bb.0:
; CHECK-NEXT:    .pad #4
; CHECK-NEXT:    sub sp, sp, #4
; CHECK-NEXT:    movw r0, #10000
; CHECK-NEXT:    strh r0, [sp, #2]
; CHECK-NEXT:    ldrh r0, [sp, #2]
; CHECK-NEXT:    add sp, sp, #4
; CHECK-NEXT:    bx lr
;
; CHECK-FP16-LABEL: bitcast_tenk:
; CHECK-FP16:       @ %bb.0:
; CHECK-FP16-NEXT:    movw r0, #10000
; CHECK-FP16-NEXT:    vmov.f16 s0, r0
; CHECK-FP16-NEXT:    vmov r0, s0
; CHECK-FP16-NEXT:    bx lr
;
; CHECK-HARD-LABEL: bitcast_tenk:
; CHECK-HARD:       @ %bb.0:
; CHECK-HARD-NEXT:    .pad #4
; CHECK-HARD-NEXT:    sub sp, sp, #4
; CHECK-HARD-NEXT:    movw r0, #10000
; CHECK-HARD-NEXT:    strh r0, [sp, #2]
; CHECK-HARD-NEXT:    ldrh r0, [sp, #2]
; CHECK-HARD-NEXT:    vmov s0, r0
; CHECK-HARD-NEXT:    add sp, sp, #4
; CHECK-HARD-NEXT:    bx lr
  %z = bitcast i16 10000 to bfloat
  ret bfloat %z
}

define bfloat @minus0() {
; CHECK-LABEL: minus0:
; CHECK:       @ %bb.0:
; CHECK-NEXT:    adr r0, .LCPI3_0
; CHECK-NEXT:    ldrh r0, [r0]
; CHECK-NEXT:    bx lr
; CHECK-NEXT:    .p2align 1
; CHECK-NEXT:  @ %bb.1:
; CHECK-NEXT:  .LCPI3_0:
; CHECK-NEXT:    .short 0x8000 @ bfloat -0
;
; CHECK-FP16-LABEL: minus0:
; CHECK-FP16:       @ %bb.0:
; CHECK-FP16-NEXT:    vldr.16 s0, .LCPI3_0
; CHECK-FP16-NEXT:    vmov r0, s0
; CHECK-FP16-NEXT:    bx lr
; CHECK-FP16-NEXT:    .p2align 1
; CHECK-FP16-NEXT:  @ %bb.1:
; CHECK-FP16-NEXT:  .LCPI3_0:
; CHECK-FP16-NEXT:    .short 0x8000 @ bfloat -0
;
; CHECK-HARD-LABEL: minus0:
; CHECK-HARD:       @ %bb.0:
; CHECK-HARD-NEXT:    adr r0, .LCPI3_0
; CHECK-HARD-NEXT:    ldrh r0, [r0]
; CHECK-HARD-NEXT:    vmov s0, r0
; CHECK-HARD-NEXT:    bx lr
; CHECK-HARD-NEXT:    .p2align 1
; CHECK-HARD-NEXT:  @ %bb.1:
; CHECK-HARD-NEXT:  .LCPI3_0:
; CHECK-HARD-NEXT:    .short 0x8000 @ bfloat -0
  ret bfloat 0xR8000
}