nir: Store gl_shader_stage in nir_shader.