llvm/test/CodeGen/AMDGPU/llvm.amdgcn.ballot.i64.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -march=amdgcn -mcpu=gfx900 < %s | FileCheck %s
   3
   4 declare i64 @llvm.amdgcn.ballot.i64(i1)
   5 declare i64 @llvm.ctpop.i64(i64)
   6
   7 ; Test ballot(0)
   8
   9 define amdgpu_cs i64 @constant_false() {
  10 ; CHECK-LABEL: constant_false:
  11 ; CHECK:       ; %bb.0:
  12 ; CHECK-NEXT:    s_mov_b32 s0, 0
  13 ; CHECK-NEXT:    s_mov_b32 s1, 0
  14 ; CHECK-NEXT:    ; return to shader part epilog
  15   %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 0)
  16   ret i64 %ballot
  17 }
  18
  19 ; Test ballot(1)
  20
  21 define amdgpu_cs i64 @constant_true() {
  22 ; CHECK-LABEL: constant_true:
  23 ; CHECK:       ; %bb.0:
  24 ; CHECK-NEXT:    s_mov_b32 s0, exec_lo
  25 ; CHECK-NEXT:    s_mov_b32 s1, exec_hi
  26 ; CHECK-NEXT:    ; return to shader part epilog
  27   %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 1)
  28   ret i64 %ballot
  29 }
  30
  31 ; Test ballot of a non-comparison operation
  32
  33 define amdgpu_cs i64 @non_compare(i32 %x) {
  34 ; CHECK-LABEL: non_compare:
  35 ; CHECK:       ; %bb.0:
  36 ; CHECK-NEXT:    v_and_b32_e32 v0, 1, v0
  37 ; CHECK-NEXT:    v_cmp_ne_u32_e64 s[0:1], 0, v0
  38 ; CHECK-NEXT:    ; return to shader part epilog
  39   %trunc = trunc i32 %x to i1
  40   %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %trunc)
  41   ret i64 %ballot
  42 }
  43
  44 ; Test ballot of comparisons
  45
  46 define amdgpu_cs i64 @compare_ints(i32 %x, i32 %y) {
  47 ; CHECK-LABEL: compare_ints:
  48 ; CHECK:       ; %bb.0:
  49 ; CHECK-NEXT:    v_cmp_eq_u32_e64 s[0:1], v0, v1
  50 ; CHECK-NEXT:    ; return to shader part epilog
  51   %cmp = icmp eq i32 %x, %y
  52   %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %cmp)
  53   ret i64 %ballot
  54 }
  55
  56 define amdgpu_cs i64 @compare_int_with_constant(i32 %x) {
  57 ; CHECK-LABEL: compare_int_with_constant:
  58 ; CHECK:       ; %bb.0:
  59 ; CHECK-NEXT:    s_movk_i32 s0, 0x62
  60 ; CHECK-NEXT:    v_cmp_lt_i32_e64 s[0:1], s0, v0
  61 ; CHECK-NEXT:    ; return to shader part epilog
  62   %cmp = icmp sge i32 %x, 99
  63   %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %cmp)
  64   ret i64 %ballot
  65 }
  66
  67 define amdgpu_cs i64 @compare_floats(float %x, float %y) {
  68 ; CHECK-LABEL: compare_floats:
  69 ; CHECK:       ; %bb.0:
  70 ; CHECK-NEXT:    v_cmp_gt_f32_e64 s[0:1], v0, v1
  71 ; CHECK-NEXT:    ; return to shader part epilog
  72   %cmp = fcmp ogt float %x, %y
  73   %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %cmp)
  74   ret i64 %ballot
  75 }
  76
  77 define amdgpu_cs i64 @ctpop_of_ballot(float %x, float %y) {
  78 ; CHECK-LABEL: ctpop_of_ballot:
  79 ; CHECK:       ; %bb.0:
  80 ; CHECK-NEXT:    v_cmp_gt_f32_e32 vcc, v0, v1
  81 ; CHECK-NEXT:    s_bcnt1_i32_b64 s0, vcc
  82 ; CHECK-NEXT:    s_mov_b32 s1, 0
  83 ; CHECK-NEXT:    ; return to shader part epilog
  84   %cmp = fcmp ogt float %x, %y
  85   %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %cmp)
  86   %bcnt = call i64 @llvm.ctpop.i64(i64 %ballot)
  87   ret i64 %bcnt
  88 }