()
| 619 | |
| 620 | |
| 621 | def test_subgraph_message_passing(): |
| 622 | # Unit test for PR #2055 |
| 623 | g = dgl.graph(([0, 1, 2], [2, 3, 4])).to(F.cpu()) |
| 624 | g.ndata["x"] = F.copy_to(F.randn((5, 6)), F.cpu()) |
| 625 | sg = g.subgraph([1, 2, 3]).to(F.ctx()) |
| 626 | sg.update_all( |
| 627 | lambda edges: {"x": edges.src["x"]}, |
| 628 | lambda nodes: {"y": F.sum(nodes.mailbox["x"], 1)}, |
| 629 | ) |
| 630 | |
| 631 | |
| 632 | @parametrize_idtype |