Add OpGroupNonUniformElect, OpGroupNonUniformBroadcastFirst

2 files changed, 19 insertions, 0 deletions

diff --git a/include/sirit/sirit.h b/include/sirit/sirit.h
index 6ced1a8..aea4468 100644
--- a/include/sirit/sirit.h
+++ b/include/sirit/sirit.h
@@ -1160,6 +1160,14 @@ public:
     /// TBD
     Id OpSubgroupAllEqualKHR(Id result_type, Id predicate);
 
+    // Result is true only in the active invocation with the lowest id in the group, otherwise
+    // result is false.
+    Id OpGroupNonUniformElect(Id result_type, Id scope);
+
+    // Result is the Value of the invocation from the active invocation with the lowest id in the
+    // group to all active invocations in the group.
+    Id OpGroupNonUniformBroadcastFirst(Id result_type, Id scope, Id value);
+
     // Result is the Value of the invocation identified by the id Id to all active invocations in
     // the group.
     Id OpGroupNonUniformBroadcast(Id result_type, Id scope, Id value, Id id);
diff --git a/src/instructions/group.cpp b/src/instructions/group.cpp
index 3b6f71a..b853284 100644
--- a/src/instructions/group.cpp
+++ b/src/instructions/group.cpp
@@ -36,6 +36,17 @@ Id Module::OpSubgroupAllEqualKHR(Id result_type, Id predicate) {
     return *code << OpId{spv::Op::OpSubgroupAllEqualKHR, result_type} << predicate << EndOp{};
 }
 
+Id Module::OpGroupNonUniformElect(Id result_type, Id scope) {
+    code->Reserve(4);
+    return *code << OpId{spv::Op::OpGroupNonUniformElect, result_type} << scope << EndOp{};
+}
+
+Id Module::OpGroupNonUniformBroadcastFirst(Id result_type, Id scope, Id value) {
+    code->Reserve(5);
+    return *code << OpId{spv::Op::OpGroupNonUniformBroadcastFirst, result_type} << scope << value
+                 << EndOp{};
+}
+
 Id Module::OpGroupNonUniformBroadcast(Id result_type, Id scope, Id value, Id id) {
     code->Reserve(6);
     return *code << OpId{spv::Op::OpGroupNonUniformBroadcast, result_type} << scope << value