import chop.passes as passes
from .auto_pipeline import AutoPipeline
[docs]
class AutoPipelineForDistributedInference(AutoPipeline):
"""This pipeline is used for distributed inference.
It runs the following passes:
- init_metadata_analysis_pass
- report_graph_analysis_pass
- add_common_metadata_analysis_pass
- autosharding_analysis_pass
- resharding_transform_pass
"""
[docs]
def __init__(self) -> None:
"""Initializes the AutoPipeline."""
pass_list = [
passes.init_metadata_analysis_pass,
passes.report_graph_analysis_pass,
passes.add_common_metadata_analysis_pass,
passes.autosharding_analysis_pass,
passes.resharding_transform_pass,
]
super().__init__(pass_list)