spirv: Add initial subgroup support