def get_next_super_layer(layer_dependency_graph, super_nodes, current_layer):
    """
    Return the immediate next super layer of current layer.

    :param layer_dependency_graph: dependency graph of the model
    :param super_nodes: list of all super nodes
    :param current_layer: the layer whose next super layer need to compute
    :return: immediate next super layer
    """
    current_layer = layer_dependency_graph[current_layer].get_output_layers()[0]
    while True:
        if current_layer in super_nodes:
            return current_layer
        current_layer = layer_dependency_graph[current_layer].get_output_layers()[0]