gallium/ureg: Set the next shader stage from the shader info.