| ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py |
| ; RUN: llc -march=amdgcn -mcpu=gfx900 -global-isel -verify-machineinstrs < %s | FileCheck %s |
| |
| declare i64 @llvm.amdgcn.ballot.i64(i1) |
| |
| ; Test ballot(0) |
| |
| define amdgpu_cs i64 @constant_false() { |
| ; CHECK-LABEL: constant_false: |
| ; CHECK: ; %bb.0: |
| ; CHECK-NEXT: s_mov_b32 s0, 0 |
| ; CHECK-NEXT: s_mov_b32 s1, 0 |
| ; CHECK-NEXT: ; return to shader part epilog |
| %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 0) |
| ret i64 %ballot |
| } |
| |
| ; Test ballot(1) |
| |
| define amdgpu_cs i64 @constant_true() { |
| ; CHECK-LABEL: constant_true: |
| ; CHECK: ; %bb.0: |
| ; CHECK-NEXT: s_mov_b32 s0, exec_lo |
| ; CHECK-NEXT: s_mov_b32 s1, exec_hi |
| ; CHECK-NEXT: ; return to shader part epilog |
| %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 1) |
| ret i64 %ballot |
| } |
| |
| ; Test ballot of a non-comparison operation |
| |
| define amdgpu_cs i64 @non_compare(i32 %x) { |
| ; CHECK-LABEL: non_compare: |
| ; CHECK: ; %bb.0: |
| ; CHECK-NEXT: v_and_b32_e32 v0, 1, v0 |
| ; CHECK-NEXT: v_cmp_ne_u32_e64 s[0:1], 0, v0 |
| ; CHECK-NEXT: ; return to shader part epilog |
| %trunc = trunc i32 %x to i1 |
| %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %trunc) |
| ret i64 %ballot |
| } |
| |
| ; Test ballot of comparisons |
| |
| define amdgpu_cs i64 @compare_ints(i32 %x, i32 %y) { |
| ; CHECK-LABEL: compare_ints: |
| ; CHECK: ; %bb.0: |
| ; CHECK-NEXT: v_cmp_eq_u32_e64 s[0:1], v0, v1 |
| ; CHECK-NEXT: ; return to shader part epilog |
| %cmp = icmp eq i32 %x, %y |
| %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %cmp) |
| ret i64 %ballot |
| } |
| |
| define amdgpu_cs i64 @compare_int_with_constant(i32 %x) { |
| ; CHECK-LABEL: compare_int_with_constant: |
| ; CHECK: ; %bb.0: |
| ; CHECK-NEXT: v_mov_b32_e32 v1, 0x63 |
| ; CHECK-NEXT: v_cmp_ge_i32_e64 s[0:1], v0, v1 |
| ; CHECK-NEXT: ; return to shader part epilog |
| %cmp = icmp sge i32 %x, 99 |
| %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %cmp) |
| ret i64 %ballot |
| } |
| |
| define amdgpu_cs i64 @compare_floats(float %x, float %y) { |
| ; CHECK-LABEL: compare_floats: |
| ; CHECK: ; %bb.0: |
| ; CHECK-NEXT: v_cmp_gt_f32_e64 s[0:1], v0, v1 |
| ; CHECK-NEXT: ; return to shader part epilog |
| %cmp = fcmp ogt float %x, %y |
| %ballot = call i64 @llvm.amdgcn.ballot.i64(i1 %cmp) |
| ret i64 %ballot |
| } |