forked from OSchip/llvm-project
74 lines
2.1 KiB
LLVM
74 lines
2.1 KiB
LLVM
|
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
|
||
|
|
; RUN: llc -march=amdgcn -mcpu=gfx900 -global-isel < %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_b64 s[0:1], exec
|
||
|
|
; 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
|
||
|
|
}
|