mesa/compiler: add stage to shader_info