xref: /llvm-project/llvm/test/CodeGen/SPIRV/transcoding/OpGroupAllAny.ll (revision 23c72e93a5af2213d616755103415bb638731203)
1; RUN: llc -O0 -mtriple=spirv64-unknown-unknown %s -o - | FileCheck %s --check-prefix=CHECK-SPIRV
2; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown %s -o - -filetype=obj | spirv-val %}
3
4; RUN: llc -O0 -mtriple=spirv32-unknown-unknown %s -o - | FileCheck %s --check-prefix=CHECK-SPIRV
5; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv32-unknown-unknown %s -o - -filetype=obj | spirv-val %}
6
7; CHECK-SPIRV: OpCapability Groups
8; CHECK-SPIRV-DAG: %[[#BoolTypeID:]] = OpTypeBool
9; CHECK-SPIRV-DAG: %[[#True:]] = OpConstantTrue %[[#BoolTypeID]]
10; CHECK-SPIRV-DAG: %[[#False:]] = OpConstantFalse %[[#BoolTypeID]]
11
12; CHECK-SPIRV: OpFunction
13; CHECK-SPIRV: %[[#]] = OpGroupAll %[[#BoolTypeID]] %[[#]] %[[#True]]
14; CHECK-SPIRV: %[[#]] = OpGroupAny %[[#BoolTypeID]] %[[#]] %[[#True]]
15; CHECK-SPIRV: %[[#]] = OpGroupAll %[[#BoolTypeID]] %[[#]] %[[#True]]
16; CHECK-SPIRV: %[[#]] = OpGroupAny %[[#BoolTypeID]] %[[#]] %[[#False]]
17define spir_kernel void @test(i32 addrspace(1)* nocapture readnone %i) {
18entry:
19  %call = tail call spir_func i32 @_Z14work_group_alli(i32 5)
20  %call1 = tail call spir_func i32 @_Z14work_group_anyi(i32 5)
21  %call3 = tail call spir_func i32 @__spirv_GroupAll(i32 0, i1 1)
22  %call4 = tail call spir_func i32 @__spirv_GroupAny(i32 0, i1 0)
23  ret void
24}
25
26; CHECK-SPIRV: OpFunction
27; CHECK-SPIRV: %[[#]] = OpGroupAny %[[#BoolTypeID]] %[[#]] %[[#]]
28; CHECK-SPIRV: %[[#]] = OpGroupAll %[[#BoolTypeID]] %[[#]] %[[#]]
29define spir_kernel void @test_nonconst_any(ptr addrspace(1) %input, ptr addrspace(1) %output)  #0 !kernel_arg_addr_space !7 !kernel_arg_access_qual !8 !kernel_arg_type !9 !kernel_arg_type_qual !10 !kernel_arg_base_type !9 !spirv.ParameterDecorations !11  {
30entry:
31  %r0 = call spir_func i64 @_Z13get_global_idj(i32 0)
32  %r1 = insertelement <3 x i64> undef, i64 %r0, i32 0
33  %r2 = call spir_func i64 @_Z13get_global_idj(i32 1)
34  %r3 = insertelement <3 x i64> %r1, i64 %r2, i32 1
35  %r4 = call spir_func i64 @_Z13get_global_idj(i32 2)
36  %r5 = insertelement <3 x i64> %r3, i64 %r4, i32 2
37  %call = extractelement <3 x i64> %r5, i32 0
38  %conv = trunc i64 %call to i32
39  %idxprom = sext i32 %conv to i64
40  %arrayidx = getelementptr inbounds float, ptr addrspace(1) %input, i64 %idxprom
41  %r6 = load float, ptr addrspace(1) %arrayidx, align 4
42  %add = add nsw i32 %conv, 1
43  %idxprom1 = sext i32 %add to i64
44  %arrayidx2 = getelementptr inbounds float, ptr addrspace(1) %input, i64 %idxprom1
45  %r7 = load float, ptr addrspace(1) %arrayidx2, align 4
46  %cmp = fcmp ogt float %r6, %r7
47  %conv3 = select i1 %cmp, i32 1, i32 0
48  %r8 = icmp ne i32 %conv3, 0
49  %r9 = zext i1 %r8 to i32
50  %r10 = call spir_func i32 @_Z14work_group_anyi(i32 %r9)
51  %call41 = icmp ne i32 %r10, 0
52  %call4 = select i1 %call41, i32 1, i32 0
53  %idxprom5 = sext i32 %conv to i64
54  %arrayidx6 = getelementptr inbounds i32, ptr addrspace(1) %output, i64 %idxprom5
55  store i32 %call4, ptr addrspace(1) %arrayidx6, align 4
56  %r11 = call spir_func i32 @_Z14work_group_alli(i32 %r9)
57  %call42 = icmp ne i32 %r11, 0
58  %call5 = select i1 %call42, i32 1, i32 0
59  store i32 %call5, ptr addrspace(1) %arrayidx6, align 4
60  ret void
61}
62
63declare spir_func i64 @_Z13get_global_idj(i32) #1
64
65declare spir_func i32 @_Z14work_group_alli(i32) #2
66declare spir_func i32 @_Z14work_group_anyi(i32) #2
67
68declare spir_func i1 @__spirv_GroupAll(i32, i1)
69declare spir_func i1 @__spirv_GroupAny(i32, i1)
70
71attributes #0 = { nounwind }
72attributes #1 = { nounwind willreturn memory(none) }
73attributes #2 = { convergent nounwind }
74
75!spirv.MemoryModel = !{!0}
76!opencl.enable.FP_CONTRACT = !{}
77!spirv.Source = !{!1}
78!opencl.spir.version = !{!2}
79!opencl.ocl.version = !{!3}
80!opencl.used.extensions = !{!4}
81!opencl.used.optional.core.features = !{!5}
82!spirv.Generator = !{!6}
83
84!0 = !{i32 2, i32 2}
85!1 = !{i32 3, i32 300000}
86!2 = !{i32 2, i32 0}
87!3 = !{i32 3, i32 0}
88!4 = !{!"cl_khr_subgroups"}
89!5 = !{}
90!6 = !{i16 6, i16 14}
91!7 = !{i32 1, i32 1}
92!8 = !{!"none", !"none"}
93!9 = !{!"float*", !"int*"}
94!10 = !{!"", !""}
95!11 = !{!5, !5}
96