subgroup: added tests for group-op instructions

Author: Firestar99

tests/ui/arch/subgroup/subgroup_non_uniform_ballot_bit_count.rs

@@ -0,0 +1,19 @@
+// build-pass
+// compile-flags: -C target-feature=+GroupNonUniform,+GroupNonUniformBallot,+ext:SPV_KHR_vulkan_memory_model
+// compile-flags: -C llvm-args=--disassemble-fn=subgroup_non_uniform_ballot_bit_count::subgroup_non_uniform_ballot_bit_count
+
+use spirv_std::arch::{GroupOperation, SubgroupMask};
+use spirv_std::spirv;
+
+unsafe fn subgroup_non_uniform_ballot_bit_count(ballot: SubgroupMask) -> u32 {
+    spirv_std::arch::subgroup_non_uniform_ballot_bit_count::<{ GroupOperation::Reduce as u32 }>(
+        ballot,
+    )
+}
+
+#[spirv(compute(threads(1, 1, 1)))]
+pub fn main() {
+    unsafe {
+        subgroup_non_uniform_ballot_bit_count(spirv_std::arch::subgroup_non_uniform_ballot(true));
+    }
+}

tests/ui/arch/subgroup/subgroup_non_uniform_ballot_bit_count.stderr

@@ -0,0 +1,8 @@
+%1 = OpFunction  %2  None %3
+%4 = OpFunctionParameter  %5
+%6 = OpLabel
+OpLine %7 493 8
+%8 = OpGroupNonUniformBallotBitCount  %2  %9 Reduce %4
+OpNoLine
+OpReturnValue %8
+OpFunctionEnd

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_clustered.rs

@@ -0,0 +1,18 @@
+// build-pass
+// compile-flags: -C target-feature=+GroupNonUniform,+GroupNonUniformArithmetic,+GroupNonUniformClustered,+ext:SPV_KHR_vulkan_memory_model
+// compile-flags: -C llvm-args=--disassemble-fn=subgroup_non_uniform_i_add_clustered::subgroup_non_uniform_i_add_clustered
+
+use glam::UVec3;
+use spirv_std::arch::{GroupOperation, SubgroupMask};
+use spirv_std::spirv;
+
+unsafe fn subgroup_non_uniform_i_add_clustered(value: u32) -> u32 {
+    spirv_std::arch::subgroup_non_uniform_i_add_clustered::<8, _>(value)
+}
+
+#[spirv(compute(threads(32, 1, 1)))]
+pub fn main(#[spirv(local_invocation_id)] local_invocation_id: UVec3) {
+    unsafe {
+        subgroup_non_uniform_i_add_clustered(local_invocation_id.x);
+    }
+}

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_clustered.stderr

@@ -0,0 +1,8 @@
+%1 = OpFunction  %2  None %3
+%4 = OpFunctionParameter  %2
+%5 = OpLabel
+OpLine %6 787 8
+%7 = OpGroupNonUniformIAdd  %2  %8 ClusteredReduce %4 %9
+OpNoLine
+OpReturnValue %7
+OpFunctionEnd

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_exclusive_scan.rs

@@ -0,0 +1,20 @@
+// build-pass
+// compile-flags: -C target-feature=+GroupNonUniform,+GroupNonUniformArithmetic,+ext:SPV_KHR_vulkan_memory_model
+// compile-flags: -C llvm-args=--disassemble-fn=subgroup_non_uniform_i_add_exclusive_scan::subgroup_non_uniform_i_add_exclusive_scan
+
+use glam::UVec3;
+use spirv_std::arch::{GroupOperation, SubgroupMask};
+use spirv_std::spirv;
+
+unsafe fn subgroup_non_uniform_i_add_exclusive_scan(value: u32) -> u32 {
+    spirv_std::arch::subgroup_non_uniform_i_add::<{ GroupOperation::ExclusiveScan as u32 }, _>(
+        value,
+    )
+}
+
+#[spirv(compute(threads(32, 1, 1)))]
+pub fn main(#[spirv(local_invocation_id)] local_invocation_id: UVec3) {
+    unsafe {
+        subgroup_non_uniform_i_add_exclusive_scan(local_invocation_id.x);
+    }
+}

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_exclusive_scan.stderr

@@ -0,0 +1,8 @@
+%1 = OpFunction  %2  None %3
+%4 = OpFunctionParameter  %2
+%5 = OpLabel
+OpLine %6 746 8
+%7 = OpGroupNonUniformIAdd  %2  %8 ExclusiveScan %4
+OpNoLine
+OpReturnValue %7
+OpFunctionEnd

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_inclusive_scan.rs

@@ -0,0 +1,20 @@
+// build-pass
+// compile-flags: -C target-feature=+GroupNonUniform,+GroupNonUniformArithmetic,+ext:SPV_KHR_vulkan_memory_model
+// compile-flags: -C llvm-args=--disassemble-fn=subgroup_non_uniform_i_add_inclusive_scan::subgroup_non_uniform_i_add_inclusive_scan
+
+use glam::UVec3;
+use spirv_std::arch::{GroupOperation, SubgroupMask};
+use spirv_std::spirv;
+
+unsafe fn subgroup_non_uniform_i_add_inclusive_scan(value: u32) -> u32 {
+    spirv_std::arch::subgroup_non_uniform_i_add::<{ GroupOperation::InclusiveScan as u32 }, _>(
+        value,
+    )
+}
+
+#[spirv(compute(threads(32, 1, 1)))]
+pub fn main(#[spirv(local_invocation_id)] local_invocation_id: UVec3) {
+    unsafe {
+        subgroup_non_uniform_i_add_inclusive_scan(local_invocation_id.x);
+    }
+}

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_inclusive_scan.stderr

@@ -0,0 +1,8 @@
+%1 = OpFunction  %2  None %3
+%4 = OpFunctionParameter  %2
+%5 = OpLabel
+OpLine %6 746 8
+%7 = OpGroupNonUniformIAdd  %2  %8 InclusiveScan %4
+OpNoLine
+OpReturnValue %7
+OpFunctionEnd

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_reduce.rs

@@ -0,0 +1,18 @@
+// build-pass
+// compile-flags: -C target-feature=+GroupNonUniform,+GroupNonUniformArithmetic,+ext:SPV_KHR_vulkan_memory_model
+// compile-flags: -C llvm-args=--disassemble-fn=subgroup_non_uniform_i_add_reduce::subgroup_non_uniform_i_add_reduce
+
+use glam::UVec3;
+use spirv_std::arch::{GroupOperation, SubgroupMask};
+use spirv_std::spirv;
+
+unsafe fn subgroup_non_uniform_i_add_reduce(value: u32) -> u32 {
+    spirv_std::arch::subgroup_non_uniform_i_add::<{ GroupOperation::Reduce as u32 }, _>(value)
+}
+
+#[spirv(compute(threads(32, 1, 1)))]
+pub fn main(#[spirv(local_invocation_id)] local_invocation_id: UVec3) {
+    unsafe {
+        subgroup_non_uniform_i_add_reduce(local_invocation_id.x);
+    }
+}

tests/ui/arch/subgroup/subgroup_non_uniform_i_add_reduce.stderr

@@ -0,0 +1,8 @@
+%1 = OpFunction  %2  None %3
+%4 = OpFunctionParameter  %2
+%5 = OpLabel
+OpLine %6 746 8
+%7 = OpGroupNonUniformIAdd  %2  %8 Reduce %4
+OpNoLine
+OpReturnValue %7
+OpFunctionEnd