130080fab1
Don't count register uses when determining the maximum number of registers used by a function. Count only the defs. This is really an underestimate of the true register usage, but in practice that's not a problem because if a function uses a register, then it has either defined it earlier, or some other function that executed before has defined it. In particular, the register counts are used: 1. When launching an entry function - in which case we're safe because the register counts of the entry function will include the register counts of all callees. 2. At function boundaries in dynamic VGPR mode. In this case it's safe because whenever we set the new VGPR allocation we take into account the outgoing_vgpr_count set by the middle-end. The main advantage of doing this is that the artificial VGPR arguments used only for preserving the inactive lanes when using the llvm.amdgcn.init.whole.wave intrinsic are no longer counted. This enables us to allocate only the registers we need in dynamic VGPR mode. --------- Co-authored-by: Thomas Symalla <5754458+tsymalla@users.noreply.github.com>
108 lines
4.0 KiB
LLVM
108 lines
4.0 KiB
LLVM
; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx803 < %s | FileCheck --check-prefixes=ALL-ASM,OSABI-AMDHSA-ASM %s
|
|
; RUN: llc -filetype=obj -mtriple=amdgcn-amd-amdhsa -mcpu=gfx803 < %s | llvm-readelf -S -r -s --notes - | FileCheck --check-prefix=OSABI-AMDHSA-ELF %s
|
|
|
|
; ALL-ASM-LABEL: {{^}}fadd:
|
|
|
|
; OSABI-AMDHSA-ASM-NOT: .hsa_code_object_version
|
|
; OSABI-AMDHSA-ASM-NOT: .hsa_code_object_isa
|
|
; OSABI-AMDHSA-ASM-NOT: .amdgpu_hsa_kernel
|
|
; OSABI-AMDHSA-ASM-NOT: .amd_kernel_code_t
|
|
|
|
; OSABI-AMDHSA-ASM: s_endpgm
|
|
; OSABI-AMDHSA-ASM: .section .rodata,"a"
|
|
; OSABI-AMDHSA-ASM: .p2align 6
|
|
; OSABI-AMDHSA-ASM: .amdhsa_kernel fadd
|
|
; OSABI-AMDHSA-ASM: .amdhsa_user_sgpr_count 12
|
|
; OSABI-AMDHSA-ASM: .amdhsa_user_sgpr_private_segment_buffer 1
|
|
; OSABI-AMDHSA-ASM: .amdhsa_user_sgpr_kernarg_segment_ptr 1
|
|
; OSABI-AMDHSA-ASM: .amdhsa_next_free_vgpr 3
|
|
; OSABI-AMDHSA-ASM: .amdhsa_next_free_sgpr 16
|
|
; OSABI-AMDHSA-ASM: .amdhsa_reserve_vcc 0
|
|
; OSABI-AMDHSA-ASM: .amdhsa_reserve_flat_scratch 0
|
|
; OSABI-AMDHSA-ASM: .end_amdhsa_kernel
|
|
; OSABI-AMDHSA-ASM: .text
|
|
|
|
; ALL-ASM-LABEL: {{^}}fsub:
|
|
|
|
; OSABI-AMDHSA-ASM-NOT: .amdgpu_hsa_kernel
|
|
; OSABI-AMDHSA-ASM-NOT: .amd_kernel_code_t
|
|
|
|
; OSABI-AMDHSA-ASM: s_endpgm
|
|
; OSABI-AMDHSA-ASM: .section .rodata,"a"
|
|
; OSABI-AMDHSA-ASM: .p2align 6
|
|
; OSABI-AMDHSA-ASM: .amdhsa_kernel fsub
|
|
; OSABI-AMDHSA-ASM: .amdhsa_user_sgpr_count 12
|
|
; OSABI-AMDHSA-ASM: .amdhsa_user_sgpr_private_segment_buffer 1
|
|
; OSABI-AMDHSA-ASM: .amdhsa_user_sgpr_kernarg_segment_ptr 1
|
|
; OSABI-AMDHSA-ASM: .amdhsa_next_free_vgpr 3
|
|
; OSABI-AMDHSA-ASM: .amdhsa_next_free_sgpr 16
|
|
; OSABI-AMDHSA-ASM: .amdhsa_reserve_vcc 0
|
|
; OSABI-AMDHSA-ASM: .amdhsa_reserve_flat_scratch 0
|
|
; OSABI-AMDHSA-ASM: .end_amdhsa_kernel
|
|
; OSABI-AMDHSA-ASM: .text
|
|
|
|
; OSABI-AMDHSA-ASM-NOT: .hsa_code_object_version
|
|
; OSABI-AMDHSA-ASM-NOT: .hsa_code_object_isa
|
|
; OSABI-AMDHSA-ASM-NOT: .amd_amdgpu_isa
|
|
; OSABI-AMDHSA-ASM-NOT: .amd_amdgpu_hsa_metadata
|
|
; OSABI-AMDHSA-ASM-NOT: .amd_amdgpu_pal_metadata
|
|
|
|
; OSABI-AMDHSA-ELF: Section Headers
|
|
; OSABI-AMDHSA-ELF: .text PROGBITS {{[0-9]+}} {{[0-9]+}} {{[0-9a-f]+}} {{[0-9]+}} AX {{[0-9]+}} {{[0-9]+}} 256
|
|
; OSABI-AMDHSA-ELF: .rodata PROGBITS {{[0-9]+}} {{[0-9]+}} {{[0-9a-f]+}} {{[0-9]+}} A {{[0-9]+}} {{[0-9]+}} 64
|
|
|
|
; OSABI-AMDHSA-ELF: Relocation section '.rela.rodata' at offset
|
|
; OSABI-AMDHSA-ELF: R_AMDGPU_REL64 0000000000000000 fadd + 10
|
|
; OSABI-AMDHSA-ELF: R_AMDGPU_REL64 0000000000000100 fsub + 10
|
|
; OSABI-AMDHSA-ELF: R_AMDGPU_REL64 0000000000000200 empty + 10
|
|
|
|
; OSABI-AMDHSA-ELF: Symbol table '.symtab' contains {{[0-9]+}} entries
|
|
; OSABI-AMDHSA-ELF: {{[0-9]+}}: 0000000000000000 {{[0-9]+}} FUNC GLOBAL PROTECTED {{[0-9]+}} fadd
|
|
; OSABI-AMDHSA-ELF: {{[0-9]+}}: 0000000000000000 64 OBJECT GLOBAL DEFAULT {{[0-9]+}} fadd.kd
|
|
; OSABI-AMDHSA-ELF: {{[0-9]+}}: 0000000000000100 {{[0-9]+}} FUNC GLOBAL PROTECTED {{[0-9]+}} fsub
|
|
; OSABI-AMDHSA-ELF: {{[0-9]+}}: 0000000000000040 64 OBJECT GLOBAL DEFAULT {{[0-9]+}} fsub.kd
|
|
|
|
; OSABI-AMDHSA-ELF: Displaying notes found in: .note
|
|
; OSABI-AMDHSA-ELF: AMDGPU 0x{{[0-9a-f]+}} NT_AMDGPU_METADATA (AMDGPU Metadata)
|
|
|
|
define amdgpu_kernel void @fadd(
|
|
ptr addrspace(1) %r,
|
|
ptr addrspace(1) %a,
|
|
ptr addrspace(1) %b) #0 {
|
|
entry:
|
|
%a.val = load float, ptr addrspace(1) %a
|
|
%b.val = load float, ptr addrspace(1) %b
|
|
%r.val = fadd float %a.val, %b.val
|
|
store float %r.val, ptr addrspace(1) %r
|
|
ret void
|
|
}
|
|
|
|
define amdgpu_kernel void @fsub(
|
|
ptr addrspace(1) %r,
|
|
ptr addrspace(1) %a,
|
|
ptr addrspace(1) %b) #0 {
|
|
entry:
|
|
%a.val = load float, ptr addrspace(1) %a
|
|
%b.val = load float, ptr addrspace(1) %b
|
|
%r.val = fsub float %a.val, %b.val
|
|
store float %r.val, ptr addrspace(1) %r
|
|
ret void
|
|
}
|
|
|
|
; Make sure kernel arguments do not count towards the number of
|
|
; registers used.
|
|
;
|
|
; ALL-ASM-LABEL: {{^}}empty:
|
|
; ALL-ASM: .amdhsa_next_free_vgpr 3
|
|
; ALL-ASM: .amdhsa_next_free_sgpr 1
|
|
define amdgpu_kernel void @empty(
|
|
i32 %i,
|
|
ptr addrspace(1) %r,
|
|
ptr addrspace(1) %a,
|
|
ptr addrspace(1) %b) #0 {
|
|
entry:
|
|
ret void
|
|
}
|
|
|
|
attributes #0 = { "amdgpu-no-flat-scratch-init" }
|