b1eeb934494ef1eee20cf2d35b718790cb9cb550,python/dgl/runtime/scheduler.py,,schedule_recv,#Any#Any#Any#Any#,55

Before Change

        def _afunc_wrapper(node_data):
            nb = NodeBatch(graph, recv_nodes, node_data)
            return apply_func(nb)
        afunc = var.FUNC(_afunc_wrapper)
        applied_feat = ir.NODE_UDF(afunc, v_nf)
        final_feat = ir.UPDATE_DICT(reduced_feat, applied_feat)
    else:
        final_feat = reduced_feat

After Change

    src, dst, mid = graph._msg_graph.in_edges(recv_nodes)
    if len(mid) == 0:
        // All recv nodes are 0-degree nodes; downgrade to apply nodes.
        if apply_func is not None:
            schedule_apply_nodes(graph, recv_nodes, apply_func)
    else:
        var_nf = var.FEAT_DICT(graph._node_frame, name="nf")
        // sort and unique the argument
        recv_nodes, _ = F.sort_1d(F.unique(recv_nodes.tousertensor()))

Instances