From 8256ee3fa363064ac3bd824d436aced81c61d23f Mon Sep 17 00:00:00 2001 From: Jason Ekstrand Date: Tue, 29 Aug 2017 09:44:44 -0700 Subject: spirv: Add subgroup shuffle support Reviewed-by: Lionel Landwerlin Reviewed-by: Iago Toral Quiroga --- src/compiler/spirv/spirv_to_nir.c | 5 +++++ src/compiler/spirv/vtn_subgroup.c | 24 +++++++++++++++++++++++- 2 files changed, 28 insertions(+), 1 deletion(-) (limited to 'src/compiler/spirv') diff --git a/src/compiler/spirv/spirv_to_nir.c b/src/compiler/spirv/spirv_to_nir.c index 1358d074993..b719ea831ff 100644 --- a/src/compiler/spirv/spirv_to_nir.c +++ b/src/compiler/spirv/spirv_to_nir.c @@ -3305,6 +3305,11 @@ vtn_handle_preamble_instruction(struct vtn_builder *b, SpvOp opcode, spv_check_supported(subgroup_ballot, cap); break; + case SpvCapabilityGroupNonUniformShuffle: + case SpvCapabilityGroupNonUniformShuffleRelative: + spv_check_supported(subgroup_shuffle, cap); + break; + case SpvCapabilityVariablePointersStorageBuffer: case SpvCapabilityVariablePointers: spv_check_supported(variable_pointers, cap); diff --git a/src/compiler/spirv/vtn_subgroup.c b/src/compiler/spirv/vtn_subgroup.c index 09e4e598b26..b999439264b 100644 --- a/src/compiler/spirv/vtn_subgroup.c +++ b/src/compiler/spirv/vtn_subgroup.c @@ -229,7 +229,29 @@ vtn_handle_subgroup(struct vtn_builder *b, SpvOp opcode, case SpvOpGroupNonUniformShuffle: case SpvOpGroupNonUniformShuffleXor: case SpvOpGroupNonUniformShuffleUp: - case SpvOpGroupNonUniformShuffleDown: + case SpvOpGroupNonUniformShuffleDown: { + nir_intrinsic_op op; + switch (opcode) { + case SpvOpGroupNonUniformShuffle: + op = nir_intrinsic_shuffle; + break; + case SpvOpGroupNonUniformShuffleXor: + op = nir_intrinsic_shuffle_xor; + break; + case SpvOpGroupNonUniformShuffleUp: + op = nir_intrinsic_shuffle_up; + break; + case SpvOpGroupNonUniformShuffleDown: + op = nir_intrinsic_shuffle_down; + break; + default: + unreachable("Invalid opcode"); + } + vtn_build_subgroup_instr(b, op, val->ssa, vtn_ssa_value(b, w[4]), + vtn_ssa_value(b, w[5])->def); + break; + } + case SpvOpGroupNonUniformIAdd: case SpvOpGroupNonUniformFAdd: case SpvOpGroupNonUniformIMul: -- cgit v1.2.3