diff options
| author | Daniel Schürmann <[email protected]> | 2018-03-09 10:55:15 +0100 |
|---|---|---|
| committer | Connor Abbott <[email protected]> | 2019-06-13 12:44:23 +0000 |
| commit | 1b89ebeedead8316bd7af0836a8890c8d1e8a131 (patch) | |
| tree | 51bb17e5f9d4bbdb045cd6976b5b77842204236d /src/compiler/spirv/vtn_subgroup.c | |
| parent | de56ebadceac8499753ba85bb712cd76168f8837 (diff) | |
nir/spirv: add support for the SubgroupBallotKHR SPIR-V capability
This capability is required for the VK_EXT_shader_subgroup_ballot extension.
Reviewed-by: Connor Abbott <[email protected]>
Diffstat (limited to 'src/compiler/spirv/vtn_subgroup.c')
| -rw-r--r-- | src/compiler/spirv/vtn_subgroup.c | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/src/compiler/spirv/vtn_subgroup.c b/src/compiler/spirv/vtn_subgroup.c index dfa8347b292..ce795ec2cb5 100644 --- a/src/compiler/spirv/vtn_subgroup.c +++ b/src/compiler/spirv/vtn_subgroup.c @@ -88,12 +88,13 @@ vtn_handle_subgroup(struct vtn_builder *b, SpvOp opcode, break; } - case SpvOpGroupNonUniformBallot: { + case SpvOpGroupNonUniformBallot: ++w; + case SpvOpSubgroupBallotKHR: { vtn_fail_if(val->type->type != glsl_vector_type(GLSL_TYPE_UINT, 4), "OpGroupNonUniformBallot must return a uvec4"); nir_intrinsic_instr *ballot = nir_intrinsic_instr_create(b->nb.shader, nir_intrinsic_ballot); - ballot->src[0] = nir_src_for_ssa(vtn_ssa_value(b, w[4])->def); + ballot->src[0] = nir_src_for_ssa(vtn_ssa_value(b, w[3])->def); nir_ssa_dest_init(&ballot->instr, &ballot->dest, 4, 32, NULL); ballot->num_components = 4; nir_builder_instr_insert(&b->nb, &ballot->instr); @@ -176,15 +177,17 @@ vtn_handle_subgroup(struct vtn_builder *b, SpvOp opcode, break; } - case SpvOpGroupNonUniformBroadcastFirst: + case SpvOpGroupNonUniformBroadcastFirst: ++w; + case SpvOpSubgroupFirstInvocationKHR: vtn_build_subgroup_instr(b, nir_intrinsic_read_first_invocation, - val->ssa, vtn_ssa_value(b, w[4]), NULL, 0, 0); + val->ssa, vtn_ssa_value(b, w[3]), NULL, 0, 0); break; - case SpvOpGroupNonUniformBroadcast: + case SpvOpGroupNonUniformBroadcast: ++w; + case SpvOpSubgroupReadInvocationKHR: vtn_build_subgroup_instr(b, nir_intrinsic_read_invocation, - val->ssa, vtn_ssa_value(b, w[4]), - vtn_ssa_value(b, w[5])->def, 0, 0); + val->ssa, vtn_ssa_value(b, w[3]), + vtn_ssa_value(b, w[4])->def, 0, 0); break; case SpvOpGroupNonUniformAll: |