项目文件夹

文件

T

文件历史

sanchit-misra fac75e166b [CPU][Kernel] Single socket spmm (#3024 )

* optimizations of spmm for CPU

* Added names of contributors

* Minor code cleanup

* Moved the spmm optimization code to a new header file

* Moved to DGL's logging method

* removed duplicate code between SpMMSumCsr and SpMMCmpCsr

* Changes made to follow Google coding style

* Fixed lint errors in spmm.h

* Fixed some lint errors from spmm_blocking_libxsmm.h

* Fixed lint errors from spmm_blocking_libxsmm.h

* Added comments to SpMMCreateLibxsmmKernel

* to enable building of tests, and other cosmetic changes

* disabling libxsmm on windows

* Put a condition to avoid opt impl for FP64 as libxsmm does not have FP64 support yet

* cosmetic changes and documentation

* cosmetic changes

* to pass lint tests

* replaced multiple allocations for buffers of indices and edges with a single allocation

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>

2021-07-13 11:18:29 +08:00

cub @ a3ee304a1f

[CUDA] Add CUDA11 support (#2308 )

2020-11-07 22:16:51 +08:00

dlpack @ bee4d1dd8d

import ffi solution from TVM

2018-09-05 10:51:31 -04:00

dmlc-core @ bfad207b44

Fix hdfs error when writing big data (#2186 )

2020-09-12 23:58:36 +08:00

googletest @ f71fb4f9a9

[Test] Add gtest to project (#547 )