4f02bb751b312aec02eab82737382f1f94829a4a,tests/pytorch/test_kvstore.py,,,#,96

Before Change

    client.shut_down()

if __name__ == "__main__":
    server_embed = th.tensor([2., 2., 2., 2., 2.])
    // use pytorch shared memory
    server_embed.share_memory_()

    pid = os.fork()

After Change

    p4 = Process(target=start_client, args=(True,))
    p5 = Process(target=start_client, args=(True,))
    p6 = Process(target=start_client, args=(False,))
    p7 = Process(target=start_client, args=(False,))


    // start server process
    p0.start()
    p1.start()
    p2.start()
    p3.start()

    // start client process
    p4.start()
    p5.start()
    p6.start()
    p7.start()


    p0.join()
    p1.join()
    p2.join()
    p3.join()

    p4.join()
    p5.join()
    p6.join()
    p7.join()

Instances