- 15 Feb, 2023 1 commit
-
-
Tingyu Wang authored
* update cugraph_relgraphconv * update equality test * update cugraph rgcn example * update RelGraphConvAgg based on latest API changes * enable fallback option to fg when fanout is large --------- Co-authored-by:Mufei Li <mufeili1996@gmail.com>
-
- 06 Dec, 2022 1 commit
-
-
Tingyu Wang authored
* update agg function with new bindings * handle optional import in __init__ * raise error in RelGraphConvAgg when pylibcugraphops not imported * Update tests/cugraph/cugraph-ops/test_cugraph_relgraphconv.py Co-authored-by:
Mufei Li <mufeili1996@gmail.com> * Update tests/cugraph/cugraph-ops/test_cugraph_relgraphconv.py Co-authored-by:
Mufei Li <mufeili1996@gmail.com> * use keyword args for readability * add missing docstring to pass CI * catch ImportError rather than ModuleNotFoundError Co-authored-by:
Mufei Li <mufeili1996@gmail.com>
-
- 18 Nov, 2022 1 commit
-
-
Tingyu Wang authored
* added relgraphconv * add tests for CuGraphRelGraphConv * move test to tests directory * update docs * address PR review 1 * address PR review, continued * remove stashing ctx.device * make fanout optional * fanout -> max_in_degree * improve docstrings consistency * initialize parameters directly in __init__ * get rid of shared memory size check * improve docs * format comments * skip test case for regularizer=None Co-authored-by:
Mufei Li <mufeili1996@gmail.com> Co-authored-by:
Hongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
-