文件历史

作者	SHA1	备注	提交日期
Tong He	972a9f1323	[Doc] Re-organize the code for dgl.geometry, and expose it in the doc (#2982 ) * reorg and expose dgl.geometry * fix lint * fix test * fix	2021-06-07 11:19:39 +08:00
Tianqi Zhang (张天启)	5d7e80f413	[Feature] Add bruteforce implementation for KNN with O(Nk) space complexity (#2892 ) * add bruteforce impl * add support for bruteforce-sharemem * modify python API * add tests * change file path * change python API * fix lint * fix test * also check worst_dist in the last few dim * use heap and early-stop on CPU * fix lint * fix lint * add device check * use cuda function to determine max shared mem * use cuda to determine block info * add memory free for tmp var * update doc-string and add dist option * fix lint * add more tests Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com> Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>	2021-05-19 18:27:43 +08:00
Tianqi Zhang (张天启)	e83d0a8007	[Feature] Add kd-tree implementation (CPU) for kNN (#2767 ) * add submodule nanoflann * finish python API for knn * finish ndarray adaptor * finish cpu-kdtree version of knn * use openmp * add endline * upt * upt * fix format and code style * upt * add warning for gpu-cpu copy * avoid contiguous copy Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com> Co-authored-by: Tong He <hetong007@gmail.com>	2021-04-09 17:39:34 +08:00
Tianqi Zhang (张天启)	c88fca5055	[Feature] Add edge coarsening for homogeneous undirected graphs (#2691 ) * finish graph matching gpu version * use C++ shuffle * finish graph matching * fix bug * fix bug * change name and use swap * upt * fix format problem * fix format problem * stronger test * upt * upt * change python api * upt * upt * format check * upt * upt * fix bug Co-authored-by: Tong He <hetong007@gmail.com>	2021-03-09 16:35:27 +08:00
Quan (Andy) Gan	2f983abe64	fix knn graph bugs (#1716 )	2020-06-28 17:31:38 +08:00
Tong He	3d47693b1f	[Op] Farthest Point Sampler in Cpp and CUDA (#1630 ) * working framework without actual algorithm logic * rename * fix * fps passes compilation * correct algorithm * add cuda implementation * update random start * before refactor * pass compilation but cuda not working * working * code working, will add docstring * add mxnet support * update docstring * update doc and test * cpplint * cpcplint * pylint * temporary fix * fix for win64 * fix unitetest * fix * fix * remove comment * move to geometry package * remove redundant include * add docstrings and comments * add proof * add validity check	2020-06-22 00:52:20 +08:00

6 次代码提交