コード例 #1
0
ファイル: network.py プロジェクト: shiyu-wangbyte/dgl
def _clear_kv_msg(garbage_msg):
    """Clear data of kvstore message
    """
    F.sync()
    for msg in garbage_msg:
        if msg.c_ptr is not None:
            _CAPI_DeleteKVMsg(msg.c_ptr)
    garbage_msg = []
コード例 #2
0
def _clear_kv_msg(msg):
    """Clear data of kvstore message

    Parameters
    ----------
    msg : KVStoreMsg
        kvstore message
    """
    if msg.data is not None:
        F.sync()
        data = F.zerocopy_to_dgl_ndarray(msg.data)
        _CAPI_DeleteNDArrayData(data)
コード例 #3
0
ファイル: network.py プロジェクト: zhilongwang/dgl-1
def _clear_kv_msg(msg):
    """Clear data of kvstore message
    """
    F.sync()
    if msg.c_ptr is not None:
        _CAPI_DeleteKVMsg(msg.c_ptr)