Function test_copy_from_gpu

tests/python/common/test_heterograph-shared-memory.py:95–101

()

    reason="Not support tensorflow for now",
)
def test_copy_from_gpu():
    hg = create_test_graph(idtype=F.int32)
    hg_gpu = hg.to(F.cuda())
    hg_share = hg_gpu.shared_memory("hg_gpu")
    p = mp.Process(target=sub_proc, args=(hg, "hg_gpu"))
    p.start()
    p.join()


# TODO: Test calling shared_memory with Blocks (a subclass of HeteroGraph)