oneflow.nn.graph.graph_config.GraphConfig.enable_auto_parallel_sbp_collector

GraphConfig.enable_auto_parallel_sbp_collector(mode: bool = True)

Use “sbp collector” to create “sbp proxy” for nodes with multiple downstream operators.