xref: /llvm-project/llvm/test/CodeGen/AMDGPU/preload-kernarg-header.ll (revision 2e5c2982819625d84e0b61aea0ec00de859f0e95)
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
2; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx942 -asm-verbose=0 < %s | FileCheck -check-prefixes=ASM %s
3; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx942 -filetype=obj < %s | llvm-objdump --arch=amdgcn --mcpu=gfx942 --disassemble - | FileCheck -check-prefixes=OBJ %s
4; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx942 -amdgpu-kernarg-preload-count=1 -asm-verbose=0 < %s | llvm-mc -triple amdgcn-amd-amdhsa -mcpu=gfx942 -filetype=obj | llvm-objdump --arch=amdgcn --mcpu=gfx942 --disassemble - | FileCheck -check-prefixes=OBJ %s
5
6; OBJ: preload_ptr_kernarg_header
7; OBJ-COUNT-60: s_nop 0
8define amdgpu_kernel void @preload_ptr_kernarg_header(ptr inreg %arg) {
9; ASM-LABEL: preload_ptr_kernarg_header:
10; ASM:         s_load_dwordx2 s[8:9], s[4:5], 0x0
11; ASM-NEXT:    s_waitcnt lgkmcnt(0)
12; ASM-NEXT:    s_branch .LBB0_0
13; ASM-NEXT:    .p2align 8
14; ASM-NEXT:  .LBB0_0:
15; ASM-NEXT:    v_mov_b64_e32 v[0:1], s[8:9]
16; ASM-NEXT:    flat_store_dwordx2 v[0:1], v[0:1]
17; ASM-NEXT:    s_endpgm
18    store ptr %arg, ptr %arg
19    ret void
20}
21
22; OBJ: preload_i32_kernarg_header
23; OBJ-COUNT-58: s_nop 0
24define amdgpu_kernel void @preload_i32_kernarg_header(ptr inreg %arg, i32 inreg %arg1) {
25; ASM-LABEL: preload_i32_kernarg_header:
26; ASM:         s_load_dwordx2 s[8:9], s[4:5], 0x0
27; ASM-NEXT:    s_load_dword s10, s[4:5], 0x8
28; ASM-NEXT:    s_waitcnt lgkmcnt(0)
29; ASM-NEXT:    s_branch .LBB1_0
30; ASM-NEXT:    .p2align 8
31; ASM-NEXT:  .LBB1_0:
32; ASM-NEXT:    v_mov_b64_e32 v[0:1], s[8:9]
33; ASM-NEXT:    v_mov_b32_e32 v2, s10
34; ASM-NEXT:    flat_store_dword v[0:1], v2
35; ASM-NEXT:    s_endpgm
36    store i32 %arg1, ptr %arg
37    ret void
38}
39
40; OBJ: non_kernel_function
41; ASM: non_kernel_function
42; OBJ-NOT: s_branch
43; ASM-NOT: s_branch
44define void @non_kernel_function(ptr %arg) {
45; ASM-LABEL: non_kernel_function:
46; ASM:         s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0)
47; ASM-NEXT:    flat_store_dwordx2 v[0:1], v[0:1]
48; ASM-NEXT:    s_waitcnt vmcnt(0) lgkmcnt(0)
49; ASM-NEXT:    s_setpc_b64 s[30:31]
50    store ptr %arg, ptr %arg
51    ret void
52}
53