• Xin Yao's avatar
    [Feature] CUDA UVA sampling for MultiLayerNeighborSampler (#3674) · 738e8318
    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
    
    * XPU_SWITCH for kDLCPUPinned
    
    * a rough version ready for testing
    
    * eliminate extra context parameter for pin/unpin
    
    * update train_sampling
    
    * fix linting
    
    * fix typo
    
    * multi-gpu uva sampling case
    
    * disable new format materialization for pinned graphs
    
    * update python doc for pin_memory_
    
    * fix unit test
    
    * UVA sampling for link prediction
    
    * dispatch most csr ops
    
    * update graphsage example to combine uva sampling and UnifiedTensor
    
    * update graphsage example to combine uva sampling and UnifiedTensor
    
    * update graphsage example to combine uva sampling and UnifiedTensor
    
    * update doc
    
    * update examples
    
    * change unitgraph and heterograph's PinMemory to in-place
    
    * update examples for multi-gpu uva sampling
    
    * update doc
    
    * fix linting
    
    * fix cpu build
    
    * fix is_pinned for DistGraph
    
    * fix is_pinned for DistGraph
    
    * update graphsage unsupervised example
    
    * update doc for gpu sampling
    
    * update some check for sampling device switching
    
    * fix linting
    
    * adapt for new dataloader
    
    * fix linting
    
    * fix
    
    * fix some name issue
    
    * adjust device check
    
    * add unit test for uva sampling & fix some zero_copy bug
    
    * fix linting
    
    * update num_threads in graphsage examples
    Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
    Co-authored-by: default avatarJinjing Zhou <VoVAllen@users.noreply.github.com>
    738e8318
array.cc 36 KB