• xiang song(charlie.song)'s avatar
    [Feature] x_dot_x builtin kernel support (#831) · 0a56d652
    xiang song(charlie.song) authored
    * upd
    
    * fig edgebatch edges
    
    * add test
    
    * trigger
    
    * Update README.md for pytorch PinSage example.
    
    Add noting that the PinSage model example under
    example/pytorch/recommendation only work with Python 3.6+
    as its dataset loader depends on stanfordnlp package
    which work only with Python 3.6+.
    
    * Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
    
    1. make dgl.nn.xxx frame agnostic
    2. make test.backend include dgl.nn modules
    3. modify test_edge_softmax of test/mxnet/test_nn.py and
        test/pytorch/test_nn.py work on both CPU and GPU
    
    * Fix style
    
    * Delete unused code
    
    * Make agnostic test only related to tests/backend
    
    1. clear all agnostic related code in dgl.nn
    2. make test_graph_conv agnostic to cpu/gpu
    
    * Fix code style
    
    * fix
    
    * doc
    
    * Make all test code under tests.mxnet/pytorch.test_nn.py
    work on both CPU and GPU.
    
    * Fix syntex
    
    * Remove rand
    
    * Start implementing masked-mm kernel.
    
    A...
    0a56d652
backward_binary_reduce_impl.cuh 13.2 KB