文件历史

作者	SHA1	备注	提交日期
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
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

5 次代码提交