文件历史

作者	SHA1	备注	提交日期
Tingyu Wang	ba21295cd5	[Model] Update `CuGraphRelGraphConv` to use new bindings from `pylibcugraphops` (#4965 ) * 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>	2022-12-06 14:00:57 +08:00
Tingyu Wang	4ecd2c8585	[Model] Add `dgl.nn.CuGraphRelGraphConv` model (#4826 ) * 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>	2022-11-18 14:04:51 +08:00
Rhett Ying	b63fc41a38	[BugFix] cugraph must be imported before torch (#4730 )	2022-10-19 10:10:01 +08:00
Hongzhi (Steve), Chen	89a4cc4db2	[Misc] Black auto fix. (#4694 ) Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>	2022-10-11 00:12:30 +08:00
Vibhu Jawa	d04534c940	[Bugfix][Hotfix] Fix cugraph Pytorch CI failures (#4622 ) * get_more_logs * Update cugraph_unit_test.sh * Update test_basics.py * Update cugraph_unit_test.sh * Update tests/cugraph/test_basics.py Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>	2022-09-23 11:49:03 +08:00
Vibhu Jawa	9ae117d3cd	[Feature] Add to_cugraph and from_cugraph to DGL (#4166 ) * Added to_cugraph and from_cugraph functionality * fix from_cugraph example * Addressed Reviews * Fix linting * Apply suggestions to docstrings from code review Co-authored-by: Minjie Wang <minjie.wang@nyu.edu> * Add API docs and remove `from_cugraph` alias * move cugraph tests to tests/cugraph/test_basics.py * remove ununsed imports from test_basics.py * remove pytest.importorskip as no longer needed Co-authored-by: Minjie Wang <minjie.wang@nyu.edu> Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>	2022-07-07 11:52:17 +08:00
Rhett Ying	85f281170f	[CI] Add new CI stage for testing cugraph (#4171 ) * [CI] add new stage specific forcuda related features based on nvidia+pytorch * build and test for gpu_nv * fix build failure * fix unit tests * make -j * install cython beforehand * copy cython lib * test cugraph tests only * fix typo * separate test script for cugraph * refactor build dgl shell	2022-07-05 10:57:07 +08:00

7 次代码提交