Function wrapper

prototype/utils/dist.py:10–13

(*args, **kwargs)

def link_dist(func):

    def wrapper(*args, **kwargs):
        dist_init()
        func(*args, **kwargs)
        dist_finalize()

    return wrapper