• Xin Yao's avatar
    [Feature] Pin dgl.graph to the page-locked memory (#3616) · 40b44a43
    Xin Yao authored
    
    
    * implement pin_memory/unpin_memory/is_pinned for dgl.graph
    
    * update python docstring
    
    * update c++ docstring
    
    * add test
    
    * fix the broken UnifiedTensor
    
    * eliminate extra context parameter for pin/unpin
    
    * fix linting
    
    * fix typo
    
    * disable new format materialization for pinned graphs
    
    * update python doc for pin_memory_
    
    * fix unit test
    
    * update doc
    
    * change unitgraph and heterograph's PinMemory to in-place
    
    * update comments for NDArray's PinMemory_ and PinData
    
    * update doc
    Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
    40b44a43
dgl.DGLGraph.rst 5.14 KB