nir: Add a pass to gather info from the shader