Function func

imperative/python/test/unit/distributed/test_distributed.py:124–125

(rank, q)

    @dist.synchronized
    def func(rank, q):
        q.put(rank)

    def worker(rank, q):
        dist.init_process_group("localhost", port, world_size, rank, rank)