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

Before Change

            reduce_func = BundledReduceFunction(reduce_func)
        self._batch_recv(u, reduce_func)
        // optional apply nodes
        self.apply_nodes(apply_node_func, u)

    def _batch_recv(self, v, reduce_func):
        if self._msg_frame.num_rows == 0:
            // no message has ever been sent

After Change

        v_is_all = is_all(v)
        if v_is_all:
            v = F.arange(0, self.number_of_nodes())
        elif isinstance(v, int):
            v = [v]
        v = utils.toindex(v)
        if len(v) == 0:
            // no vertex to be triggered.
            return

        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)

Instances