nir/spirv: Properly assign locations to split structures