- 19 May, 2021 1 commit
-
-
Tianqi Zhang (张天启) authored
* 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>
-
- 09 Apr, 2021 1 commit
-
-
Tianqi Zhang (张天启) authored
* 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>
-