test_ipc.py

import dgl
import torch as th
import torch.multiprocessing as mp
import os
import unittest

def sub_ipc(g):
    print(g)
    return g

@unittest.skipIf(os.name == 'nt', reason='Do not support windows yet')
def test_torch_ipc():
    g = dgl.graph([(0, 1), (1, 2), (2, 3)])
    ctx = mp.get_context("spawn")
    p = ctx.Process(target=sub_ipc, args=(g, ))

    p.start()
    p.join()


if __name__ == "__main__":
    test_torch_ipc()