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()
    if pid == 0:
        start_server(server_embed)
    else:
        time.sleep(2) // wait server start
        start_client(server_embed)

    assert th.equal(server_embed, th.tensor([ 4., 4., 14., 4., 24.])) == True

After Change

    // server process
    p0 = Process(target=start_server, args=(0, 4))
    p1 = Process(target=start_server, args=(1, 4))
    p2 = Process(target=start_server, args=(2, 4))
    p3 = Process(target=start_server, args=(3, 4))

    // client process
    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()

Instances