xref: /llvm-project/llvm/test/CodeGen/AMDGPU/s_addk_i32.ll (revision b1bcb7ca460fcd317bbc8309e14c8761bf8394e0)
1; RUN: llc -mtriple=amdgcn--amdpal -mcpu=tahiti -verify-machineinstrs < %s | FileCheck -check-prefix=SI %s
2; RUN: llc -mtriple=amdgcn--amdpal -mcpu=tonga -mattr=-flat-for-global,-xnack -verify-machineinstrs < %s | FileCheck -check-prefix=SI %s
3
4; TODO: Some of those tests fail with OS == amdhsa due to unreasonable register
5;       allocation differences.
6
7; SI-LABEL: {{^}}s_addk_i32_k0:
8; SI: s_load_dword [[VAL:s[0-9]+]]
9; SI: s_addk_i32 [[VAL]], 0x41
10; SI: v_mov_b32_e32 [[VRESULT:v[0-9]+]], [[VAL]]
11; SI: buffer_store_dword [[VRESULT]]
12; SI: s_endpgm
13define amdgpu_kernel void @s_addk_i32_k0(ptr addrspace(1) %out, i32 %b) #0 {
14  %add = add i32 %b, 65
15  store i32 %add, ptr addrspace(1) %out
16  ret void
17}
18
19; SI-LABEL: {{^}}s_addk_i32_k0_x2:
20; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
21; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
22; SI: s_endpgm
23define amdgpu_kernel void @s_addk_i32_k0_x2(ptr addrspace(1) %out0, ptr addrspace(1) %out1, i32 %a, i32 %b) #0 {
24  %add0 = add i32 %a, 65
25  %add1 = add i32 %b, 65
26  store i32 %add0, ptr addrspace(1) %out0
27  store i32 %add1, ptr addrspace(1) %out1
28  ret void
29}
30
31; SI-LABEL: {{^}}s_addk_i32_k1:
32; SI: s_addk_i32 {{s[0-9]+}}, 0x7fff{{$}}
33; SI: s_endpgm
34define amdgpu_kernel void @s_addk_i32_k1(ptr addrspace(1) %out, i32 %b) #0 {
35  %add = add i32 %b, 32767 ; (1 << 15) - 1
36  store i32 %add, ptr addrspace(1) %out
37  ret void
38}
39
40; SI-LABEL: {{^}}s_addk_i32_k2:
41; SI: s_sub_i32 s{{[0-9]+}}, s{{[0-9]+}}, 17
42; SI: s_endpgm
43define amdgpu_kernel void @s_addk_i32_k2(ptr addrspace(1) %out, i32 %b) #0 {
44  %add = add i32 %b, -17
45  store i32 %add, ptr addrspace(1) %out
46  ret void
47}
48
49; SI-LABEL: {{^}}s_addk_i32_k3:
50; SI: s_addk_i32 {{s[0-9]+}}, 0xffbf{{$}}
51; SI: s_endpgm
52define amdgpu_kernel void @s_addk_i32_k3(ptr addrspace(1) %out, i32 %b) #0 {
53  %add = add i32 %b, -65
54  store i32 %add, ptr addrspace(1) %out
55  ret void
56}
57
58; SI-LABEL: {{^}}s_addk_v2i32_k0:
59; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
60; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x42
61; SI: s_endpgm
62; Note: dummy argument here to prevent combining of descriptor loads for %out and %b
63define amdgpu_kernel void @s_addk_v2i32_k0(ptr addrspace(1) %out, i32 %dummy, <2 x i32> %b) #0 {
64  %add = add <2 x i32> %b, <i32 65, i32 66>
65  store <2 x i32> %add, ptr addrspace(1) %out
66  ret void
67}
68
69; SI-LABEL: {{^}}s_addk_v4i32_k0:
70; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
71; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x42
72; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x43
73; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x44
74; SI: s_endpgm
75define amdgpu_kernel void @s_addk_v4i32_k0(ptr addrspace(1) %out, <4 x i32> %b) #0 {
76  %add = add <4 x i32> %b, <i32 65, i32 66, i32 67, i32 68>
77  store <4 x i32> %add, ptr addrspace(1) %out
78  ret void
79}
80
81; SI-LABEL: {{^}}s_addk_v8i32_k0:
82; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x41
83; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x42
84; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x43
85; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x44
86; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x45
87; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x46
88; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x47
89; SI-DAG: s_addk_i32 {{s[0-9]+}}, 0x48
90; SI: s_endpgm
91define amdgpu_kernel void @s_addk_v8i32_k0(ptr addrspace(1) %out, <8 x i32> %b) #0 {
92  %add = add <8 x i32> %b, <i32 65, i32 66, i32 67, i32 68, i32 69, i32 70, i32 71, i32 72>
93  store <8 x i32> %add, ptr addrspace(1) %out
94  ret void
95}
96
97; SI-LABEL: {{^}}no_s_addk_i32_k0:
98; SI: s_add_i32 {{s[0-9]+}}, {{s[0-9]+}}, 0x8000{{$}}
99; SI: s_endpgm
100define amdgpu_kernel void @no_s_addk_i32_k0(ptr addrspace(1) %out, i32 %b) #0 {
101  %add = add i32 %b, 32768 ; 1 << 15
102  store i32 %add, ptr addrspace(1) %out
103  ret void
104}
105
106@lds = addrspace(3) global [512 x i32] undef, align 4
107
108; SI-LABEL: {{^}}commute_s_addk_i32:
109; SI: s_addk_i32 s{{[0-9]+}}, 0x800{{$}}
110define amdgpu_kernel void @commute_s_addk_i32(ptr addrspace(1) %out, i32 %b) #0 {
111  %size = call i32 @llvm.amdgcn.groupstaticsize()
112  %add = add i32 %size, %b
113  call void asm sideeffect "; foo $0, $1", "v,s"(ptr addrspace(3) @lds, i32 %add)
114  ret void
115}
116
117declare i32 @llvm.amdgcn.groupstaticsize() #1
118
119attributes #0 = { nounwind "amdgpu-no-dispatch-id" "amdgpu-no-dispatch-ptr" "amdgpu-no-implicitarg-ptr" "amdgpu-no-lds-kernel-id" "amdgpu-no-queue-ptr" "amdgpu-no-workgroup-id-x" "amdgpu-no-workgroup-id-y" "amdgpu-no-workgroup-id-z" "amdgpu-no-workitem-id-x" "amdgpu-no-workitem-id-y" "amdgpu-no-workitem-id-z" }
120attributes #1 = { nounwind readnone }
121