文件历史

作者	SHA1	备注	提交日期
Tingyu Wang	ea706cae58	[Model] Update cugraph-ops models for 23.04 release (#5540 ) Co-authored-by: schmidt-ju <79865721+schmidt-ju@users.noreply.github.com> Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com> Co-authored-by: Mufei Li <mufeili1996@gmail.com> Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal> Co-authored-by: peizhou001 <110809584+peizhou001@users.noreply.github.com> Co-authored-by: Ubuntu <ubuntu@ip-172-31-16-19.ap-northeast-1.compute.internal>	2023-05-04 17:11:22 +08:00
Hongzhi (Steve), Chen	5b409bf7f5	Rename_test (#5487 ) Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>	2023-03-24 18:44:24 +08:00
Tingyu Wang	bfd411d0d5	[Model] Add `dgl.nn.CuGraphGATConv` model (#5168 ) * add CuGraphGATConv model * lintrunner * update model to reflect changes in make_mfg_csr(), move max_in_degree to forward() * simplify pytest markers * fall back to FG option for large fanout * update error msg * add feat_drop and activation options * add residual option * Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * reset res_fc --------- Co-authored-by: Mufei Li <mufeili1996@gmail.com>	2023-03-04 14:10:06 +08:00
Tingyu Wang	bcf9923b32	[Model] Add `dgl.nn.CuGraphSAGEConv` model (#5137 ) * add CuGraphSAGEConv model * fix lint issues * update model to reflect changes in make_mfg_csr(), move max_in_degree to forward() * lintrunner * allow reset_parameters() * remove norm option, simplify test * allow full graph fallback option, add example * address comments * address reviews --------- Co-authored-by: Mufei Li <mufeili1996@gmail.com>	2023-02-22 13:43:28 +08:00
Hongzhi (Steve), Chen	f0759a96ee	[Misc] Auto-reformat test/. (#5324 ) * auto-format-test * more * remove --------- Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>	2023-02-19 22:50:37 +08:00
Tingyu Wang	19b3cea166	[Model] Update `CuGraphRelGraphConv` to use `pylibcugraphops=23.02` (#5217 ) * 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>	2023-02-15 20:11:20 +08:00
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

13 次代码提交