oneflow.nn.graph.graph_config.GraphConfig.enable_auto_parallel_trunk_algo

GraphConfig.enable_auto_parallel_trunk_algo(mode: bool = True)

Find the trunk of the SBP graph, then reduce the wait time for tributaries.