Update emit_spirv_warp.cpp

This commit is contained in:
DanielSvoboda 2024-07-28 23:11:17 -03:00 committed by GitHub
parent 953fd5cb76
commit 41912d0a63
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -14,6 +14,10 @@ Id EmitLaneId(EmitContext& ctx) {
return ctx.OpLoad(ctx.U32[1], ctx.subgroup_local_invocation_id); return ctx.OpLoad(ctx.U32[1], ctx.subgroup_local_invocation_id);
} }
Id EmitWarpId(EmitContext& ctx) {
return ctx.OpLoad(ctx.U32[1], ctx.subgroup_id);
}
Id EmitQuadShuffle(EmitContext& ctx, Id value, Id index) { Id EmitQuadShuffle(EmitContext& ctx, Id value, Id index) {
return ctx.OpGroupNonUniformQuadBroadcast(ctx.U32[1], SubgroupScope(ctx), value, index); return ctx.OpGroupNonUniformQuadBroadcast(ctx.U32[1], SubgroupScope(ctx), value, index);
} }