Function copy_data_to_shared_memory

python/dgl/distributed/rpc.py:1182–1184

Copy tensor data to shared-memory tensor

(dst, source)

def copy_data_to_shared_memory(dst, source):
    """Copy tensor data to shared-memory tensor"""
    F.zerocopy_to_dgl_ndarray(dst).copyfrom(F.zerocopy_to_dgl_ndarray(source))


############### Some basic services will be defined here #############