llvm/test/CodeGen/AMDGPU/divergence-driven-ctpop.ll

   1 ; RUN: llc -march=amdgcn -stop-after=amdgpu-isel < %s | FileCheck -check-prefix=GCN %s
   2
   3 ; GCN-LABEL: name:            s_ctpop_i32
   4 ; GCN: S_BCNT1_I32_B32
   5 define amdgpu_kernel void @s_ctpop_i32(ptr addrspace(1) noalias %out, i32 %val) nounwind {
   6   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
   7   store i32 %ctpop, ptr addrspace(1) %out, align 4
   8   ret void
   9 }
  10
  11 ; GCN-LABEL: name:            s_ctpop_i64
  12 ; GCN: %[[BCNT:[0-9]+]]:sreg_32 = S_BCNT1_I32_B64
  13 ; GCN: %[[SREG1:[0-9]+]]:sreg_32 = COPY %[[BCNT]]
  14 ; GCN: %[[SREG2:[0-9]+]]:sreg_32 = S_MOV_B32 0
  15 ; GCN: REG_SEQUENCE killed %[[SREG1]], %subreg.sub0, killed %[[SREG2]], %subreg.sub1
  16 define amdgpu_kernel void @s_ctpop_i64(ptr addrspace(1) noalias %out, i64 %val) nounwind {
  17   %ctpop = call i64 @llvm.ctpop.i64(i64 %val) nounwind readnone
  18   %truncctpop = trunc i64 %ctpop to i32
  19   store i32 %truncctpop, ptr addrspace(1) %out, align 4
  20   ret void
  21 }
  22
  23 ; GCN-LABEL: name:            v_ctpop_i32
  24 ; GCN: V_BCNT_U32_B32_e64
  25 define amdgpu_kernel void @v_ctpop_i32(ptr addrspace(1) noalias %out, ptr addrspace(1) noalias %in) nounwind {
  26   %tid = call i32 @llvm.amdgcn.workitem.id.x()
  27   %in.gep = getelementptr i32, ptr addrspace(1) %in, i32 %tid
  28   %val = load i32, ptr addrspace(1) %in.gep, align 4
  29   %ctpop = call i32 @llvm.ctpop.i32(i32 %val) nounwind readnone
  30   store i32 %ctpop, ptr addrspace(1) %out, align 4
  31   ret void
  32 }
  33
  34 ; GCN-LABEL: name:            v_ctpop_i64
  35 ; GCN: %[[BCNT1:[0-9]+]]:vgpr_32 = V_BCNT_U32_B32_e64 killed %{{[0-9]+}}, 0, implicit $exec
  36 ; GCN: %[[BCNT2:[0-9]+]]:vgpr_32 = V_BCNT_U32_B32_e64 killed %{{[0-9]+}}, killed %[[BCNT1]], implicit $exec
  37 ; GCN: %[[VGPR1:[0-9]+]]:vgpr_32 = V_MOV_B32_e32 0, implicit $exec
  38 ; GCN: REG_SEQUENCE killed %[[BCNT2]], %subreg.sub0, killed %[[VGPR1]], %subreg.sub1
  39 define amdgpu_kernel void @v_ctpop_i64(ptr addrspace(1) noalias %out, ptr addrspace(1) noalias %in) nounwind {
  40   %tid = call i32 @llvm.amdgcn.workitem.id.x()
  41   %in.gep = getelementptr i64, ptr addrspace(1) %in, i32 %tid
  42   %val = load i64, ptr addrspace(1) %in.gep, align 8
  43   %ctpop = call i64 @llvm.ctpop.i64(i64 %val) nounwind readnone
  44   %truncctpop = trunc i64 %ctpop to i32
  45   store i32 %truncctpop, ptr addrspace(1) %out, align 4
  46   ret void
  47 }
  48
  49 declare i64 @llvm.ctpop.i64(i64) nounwind readnone
  50
  51 declare i32 @llvm.ctpop.i32(i32) nounwind readnone
  52
  53 declare i32 @llvm.amdgcn.workitem.id.x() nounwind readnone