mirrors / dmlc--dgl

项目文件夹

dmlc--dgl
文件
0.6.x
T
文件历史
Quan (Andy) Gan 929d863447 [Feature] Sparse-sparse matrix multiplication, addition, and masking (#2753) 
* test

* more stuff

* add test

* fixes

* optimize algo

* replace unordered_map with arrays

* lint

* lint x2

* oops

* disable gpu csrmm tests

* remove gpu invocation

* optimize with openmp

* remove python functions

* add back with docstrings

* lint

* lint

* update python interface

* functionize

* functionize

* lint

* lint
2021-03-24 17:24:41 +08:00
..
common.h
[CUDA][Kernel] More CUDA kernels; Standardize the behavior for sorted COO/CSR (#1704)
2020-06-28 18:37:28 +08:00
graph_index_test.cc
[Feature] Deprecate multigraph (#1389)
2020-03-28 21:31:24 +08:00
message_queue_test.cc
[RPC] New RPC infrastructure. (#1549)
2020-05-22 18:48:37 +08:00
socket_communicator_test.cc
[RPC] Client will retry connection until the server starts. (#1758)
2020-07-07 13:34:41 +08:00
string_test.cc
[RPC] New RPC infrastructure. (#1549)
2020-05-22 18:48:37 +08:00
test_aten.cc
[CUDA][Kernel] A bunch of int64 kernels for COO and CSR (#1883)
2020-07-30 21:02:07 +08:00
test_csrmm.cc
[Feature] Sparse-sparse matrix multiplication, addition, and masking (#2753)
2021-03-24 17:24:41 +08:00
test_rowwise.cc
[Windows] Work around the bug from CL 19.10 (#2059)
2020-08-19 15:24:46 +08:00
test_sampler.cc
[Windows] Work around the bug from CL 19.10 (#2059)
2020-08-19 15:24:46 +08:00
test_serialize.cc
[Feature] Sparse-sparse matrix multiplication, addition, and masking (#2753)
2021-03-24 17:24:41 +08:00
test_smart_ptr_serialize.cc
[Feature] Support serialization for smart pointer (#1291)
2020-02-26 17:29:27 +08:00
test_spmat_coo.cc
[CUDA][Kernel] A bunch of int64 kernels for COO and CSR (#1883)
2020-07-30 21:02:07 +08:00
test_spmat_csr.cc
[Performance] Improve COO to CSR, and sort columns of CSR only when necessary. (#2391)
2021-01-27 15:33:34 +08:00
test_spmm.cc
[Feature] Tvm integration (#2367)
2020-12-31 17:40:25 +08:00
test_unit_graph.cc
[Feature] Sparse-sparse matrix multiplication, addition, and masking (#2753)
2021-03-24 17:24:41 +08:00
test_zerocopy_serialize.cc
[RPC] Support null NDArray in RPC communication (#1653)
2020-06-21 17:20:58 +08:00