def _root_rank_select(old_val, new_val): # pragma: no cover
if distributed_api.get_rank() == 0:
return old_val
return new_val
def _root_rank_select(old_val, new_val):
if distributed_api.get_rank() == 0:
return old_val
return new_val