nir: add callback to nir_remove_dead_variables()