- 15 Dec, 2021 1 commit
-
-
lixiaobai authored
* Feat: support API "randomwalk_topk" in library * Feat: use the new API "randomwalk_topk" for PinSAGESampler * Minor * Minor * Refactor: modified codes as checker required * Minor * Minor * Minor * Minor * Fix: checking errors in RandomWalkTopk * Refactor: modified the docstring for randomwalk_topk * change randomwalk_topk to internal * fix * rename * Minor for pinsage.py * Feat: support randomwalk and SelectPinSageNeighbors on GPU Port RandomWalk algorithm on GPU, and port SelectPinSageNeighbors on GPU. * Feat: support GPU on python APIs * Feat: remove perf print information in FrequenchHashmap * Fix: modified the code format Modified the code format as task_lint.sh suggested * Feat: let test script support PinSAGESampler on GPU Let test script support PinSAGESampler on GPU, minor of "restart_prob". * Minor * Minor * Minor * Refactor: use the atomic operations from the array module * Minor: change the long lines * Refactor: modified the get_node_types for gpu * Feat: update the contributor date * Perf: remove unnecessary stream sync * Feat: support other random walk But the non-uniform choice is still not supported. * Fix: add CUDA switch for random walk Co-authored-by:Quan Gan <coin2028@hotmail.com>
-
- 29 Nov, 2021 1 commit
-
-
lixiaobai authored
* Feat: support API "randomwalk_topk" in library * Feat: use the new API "randomwalk_topk" for PinSAGESampler * Minor * Minor * Refactor: modified codes as checker required * Minor * Minor * Minor * Minor * Fix: checking errors in RandomWalkTopk * Refactor: modified the docstring for randomwalk_topk * change randomwalk_topk to internal * fix * rename * Minor for pinsage.py Co-authored-by:Quan Gan <coin2028@hotmail.com>
-
- 23 Jun, 2021 1 commit
-
-
Quan (Andy) Gan authored
* add seal example * 1. add paper infomation in examples/README 2. adjust codes 3. option test * use latest `to_simple` to replace coalesce graph function * remove outdated codes * remove useless comment * Node2vec 1.implement node2vec random walk c++ op 2.implement node2vec model 3.implement node2vec example * add CMakeLists file modify * refine c++ codes * refine c++ codes * add missing whitespace * refine python codes * add codes * add node2vec_impl.h * fix codes * fix code style problem * fixes * remove * lots of changes * add benchmark * fixes Co-authored-by:
smilexuhc <smile.xuhc@gmail.com> Co-authored-by:
Minjie Wang <wmjlyjemaine@gmail.com>
-
- 15 Jun, 2020 1 commit
-
-
Minjie Wang authored
* add cuda source * moving codes from kernel2 branch * operator overloading * Better error message for unsupported device * fix c tests * coo sort using cusparse * move test_rpc to distributed * lint * address comments and add utests Co-authored-by:
Jinjing Zhou <VoVAllen@users.noreply.github.com> Co-authored-by:
Chao Ma <mctt90@gmail.com> Co-authored-by:
xiang song(charlie.song) <classicxsong@gmail.com>
-
- 28 Feb, 2020 1 commit
-
-
Quan (Andy) Gan authored
* [WIP] PinSAGE operators * moved the edge remapping mess into C * some docstrings * lint * lint x2 * lint x3 * skip gpu test on topk * extend pinsage to any metapath * lint x4 * addresses #1265 * add always_preserve (fixes #1266) and fix a silly bug * disable gpu test on compaction * lint * fix a horrible bug and add more tests * lint * addresses comments * lint * bugfix * addresses comments Co-authored-by:Minjie Wang <minjie.wang@nyu.edu>
-
- 17 Feb, 2020 1 commit
-
-
Minjie Wang authored
* WIP: working on random choices * light slice * basic CPU impl * add python binding; fix CreateFromCOO and CreateFromCSR returning unitgraph * simple test case works * fix bug in slicing probability array * fix bug in getting the correct relation graph * fix bug in creating placeholder graph * enable omp * add cpp test * sample topk * add in|out_subgraph * try fix lint; passed all unittests * fix lint * fix msvc compile; add sorted flag and constructors * fix msvc * coosort * COOSort; CSRRowWiseSampling; CSRRowWiseTopk * WIP: remove DType in CSR and COO; Restrict data array to be IdArray * fix all CSR ops for missing data array * compiled * passed tests * lint * test sampling out edge * test different per-relation fanout/k values * fix bug in random choice * finished cpptest * fix compile * Add induced edges * add check * fixed bug in sampling on hypersparse graph; add tests * add ascending flag * in|out_subgraph returns subgraph and induced eid * address comments * lint * fix
-
- 04 Feb, 2020 1 commit
-
-
Quan (Andy) Gan authored
* unit graph that prefers coo queries * auto detect coo preference * forgot some functions * disable lint on detect_prefer_coo * reorg * change comment * lint * fix * move array_utils.h to src * compact graph impl * fix redundant copying in idhashmap * docstring * moving preference detection to C * lint * fix unit test & address comments * hypersparse autorestrict * docstring & fix * revert copyto and asnumbits * fix stupid bug * lint * leave a TODO for sorted COO * fixing same node type mapping to different id in different graphs * addresses comments * made induced nodes a feautre column * lint?
-
- 31 Jan, 2020 1 commit
-
-
Quan (Andy) Gan authored
* trying to refactor IndexSelect * partial implementation * add index select and assign for floats as well * move to random choice source * more updates * fixes * fixes * more fixes * adding python impl * fixes * unit test * lint * lint x2 * lint x3 * update metapath2vec * debugging performance * still debugging for performance * tuning * switching to succvec * redo * revert non-uniform sampler to use vector * still not fast * why does this crash with OpenMP??? * because there was a data race!!! * add documentations and remove assign op * lint * lint x2 * lol what have i done * lint x3 * fix and disable gpu testing * bugfix * generic random walk * reorg the random walk source code * Update randomwalks.h * Update randomwalks_cpu.cc * rename file * move internal function to anonymous ns * reorg & docstrings * constant restart probability * docstring fix * more commit * random walk with restart, tested * some fixes * switch to using NDArray for choice * massive fix & docstring * lint x? * lint x?? * fix * export symbols * skip gpu test * addresses comments * replaces another VecToIdArray * add randomwalks.h to include * replace void * with template
-