• nv-dlasalle's avatar
    [Performance][GPU] Enable GPU uniform edge sampling (#2716) · e70138bb
    nv-dlasalle authored
    
    
    * Start on uniform GPU sampling
    
    * Save more work
    
    * Get cu file compiling
    
    * Update sampling
    
    * More changes
    
    * Get GPU sampling for uniform probabilities solved
    
    * Fix batch tensor migration
    
    * Fix
    
    * update kernels
    
    * expand blocking
    
    * Undo testing change
    
    * Cut down on sampling overhead
    
    * Fix replacement
    
    * Update unit tests
    
    * Add option to gpu sample in graphsage
    
    * Copy only csc to gpu
    
    * Add ogbn support
    
    * Fix linting
    
    * Remove nvtx from sample
    
    * Improve documentation and error checking
    
    * Expand documentation
    
    * Update assert checking
    
    * delete extra space
    
    * Use standard dataloader when dataset is a dictionary
    
    * ogb -> ogbn
    
    * Fix edge selection determinism
    
    * Fix typos
    
    * Remove nvtx
    
    * Add comment for self.fanout_arrays and assert
    
    * Fix linting
    
    * Migrate to scalarbatcher
    
    * Fix indentation
    
    * Fix batcher
    
    * Fix indexing
    
    * Only use databatcher for GPU
    
    * Convert to DGL NDArray to PyTorch Tensor
    
    * Add optimization for PyTorch's F.tensor() for list of GPU tensors
    Co-authored-by: default avatarDa Zheng <zhengda1936@gmail.com>
    e70138bb
train_sampling.py 9.99 KB