• Minjie Wang's avatar
    [CUDA] Many CUDA operators; Prepare for DGLGraph on CUDA (#1660) · f1b19a6b
    Minjie Wang authored
    * add cuda utils; change g.to; add g.device
    
    * split array.h into several headers
    
    * cuda index select
    
    * file
    
    * three cuda kernels
    
    * add cuda elementwise arith and several others
    
    * cuda CSRIsNonZero
    
    * fix lint
    
    * lint
    
    * lint
    
    * fix bug in changing ctx to property
    
    * address comments
    
    * remove unused codes
    
    * address comments
    f1b19a6b
array_index_select.cu 2.79 KB