nir: Move new edgeflag assert into the io_lowered case