forked from OSchip/llvm-project
22 lines
1.0 KiB
LLVM
22 lines
1.0 KiB
LLVM
|
; RUN: llc -mtriple amdgcn-amdhsa -mcpu=fiji -amdgpu-scalarize-global-loads -verify-machineinstrs < %s | FileCheck -check-prefix=GCN %s
|
||
|
|
||
|
declare i32 @llvm.amdgcn.workitem.id.x()
|
||
|
declare i32 @llvm.amdgcn.readfirstlane(i32)
|
||
|
|
||
|
; GCN-LABEL: readfirstlane_uniform
|
||
|
; GCN: s_load_dwordx2 s{{\[}}[[IN_ADDR:[0-9]+]]:1{{\]}}, s[4:5], 0x0
|
||
|
; GCN: v_readfirstlane_b32 s[[SCALAR:[0-9]+]], v0
|
||
|
; GCN: s_add_u32 s[[LOAD_ADDR:[0-9]+]], s[[IN_ADDR]], s[[SCALAR]]
|
||
|
; GCN: s_load_dword s{{[0-9]+}}, s{{\[}}[[LOAD_ADDR]]
|
||
|
|
||
|
define amdgpu_kernel void @readfirstlane_uniform(float addrspace(1)* noalias nocapture readonly, float addrspace(1)* noalias nocapture readonly) {
|
||
|
%tid = tail call i32 @llvm.amdgcn.workitem.id.x()
|
||
|
%scalar = tail call i32 @llvm.amdgcn.readfirstlane(i32 %tid)
|
||
|
%idx = zext i32 %scalar to i64
|
||
|
%gep0 = getelementptr inbounds float, float addrspace(1)* %0, i64 %idx
|
||
|
%val = load float, float addrspace(1)* %gep0, align 4
|
||
|
%gep1 = getelementptr inbounds float, float addrspace(1)* %1, i64 10
|
||
|
store float %val, float addrspace(1)* %gep1, align 4
|
||
|
ret void
|
||
|
}
|