fixed nvidia proprietary hack

Author: Hugobros3

src/shady/emit/spirv/emit_spv_instructions.c

@@ -5,6 +5,7 @@
 
 #include "../../type.h"
 #include "../../transform/memory_layout.h"
+#include "../../transform/ir_gen_helpers.h"
 
 #include <assert.h>
 
@@ -236,8 +237,8 @@ static void emit_primop(Emitter* emitter, FnBuilder fn_builder, BBBuilder bb_bui
 
             if (emitter->configuration->hacks.spv_shuffle_instead_of_broadcast_first) {
                 SpvId local_id;
-                error("TODO: fix")
-                //emit_primop(emitter, fn_builder, bb_builder, prim_op(emitter->arena, (PrimOp) { .op = subgroup_local_id_op }), 1, &local_id);
+                const Node* b = ref_decl_helper(emitter->arena, get_builtin(emitter->module, BuiltinSubgroupLocalInvocationId, NULL));
+                emit_primop(emitter, fn_builder, bb_builder, prim_op(emitter->arena, (PrimOp) { .op = load_op, .operands = singleton(b) }), 1, &local_id);
                 result = spvb_group_shuffle(bb_builder, emit_type(emitter, get_unqualified_type(first(args)->type)), scope_subgroup, emit_value(emitter, bb_builder, first(args)), local_id);
                 spvb_capability(emitter->file_builder, SpvCapabilityGroupNonUniformShuffle);
             } else {