deb653f8dc9b396ba99d5c66e0658fb056f37a38,python/dgl/graph.py,DGLGraph,push,#DGLGraph#Any#Any#Any#Any#,1305

Before Change

        u = utils.toindex(u)
        if len(u) == 0:
            return
        uu, vv, _ = self._graph.out_edges(u)
        self.send_and_recv((uu, vv), message_func,
                reduce_func, apply_node_func)

After Change

        u = utils.toindex(u)
        if len(u) == 0:
            return
        with ir.prog() as prog:
            scheduler.schedule_push(graph=self, u=u,
                    message_func=message_func, reduce_func=reduce_func,
                    apply_func=apply_node_func)
            Runtime.run(prog)

    def update_all(self,
                   message_func="default",
                   reduce_func="default",
                   apply_node_func="default"):

Instances