nir/gather_info: Add more info for geometry shaders