nir/spirv: Add support for device groups