deb653f8dc9b396ba99d5c66e0658fb056f37a38,python/dgl/graph.py,DGLGraph,recv,#DGLGraph#Any#Any#Any#,1106

Before Change

        if reduce_func == "default":
            reduce_func = self._reduce_func
        assert reduce_func is not None
        if isinstance(reduce_func, (list, tuple)):
            reduce_func = BundledReduceFunction(reduce_func)
        self._batch_recv(u, reduce_func)
        // optional apply nodes
        self.apply_nodes(apply_node_func, u)

After Change

        with ir.prog() as prog:
            scheduler.schedule_recv(graph=self, recv_nodes=v,
                    reduce_func=reduce_func, apply_func=apply_node_func)
            Runtime.run(prog)

        // FIXME(minjie): multi send bug
        self.reset_messages()

Instances