CompilerConfig类
该类用于构造传入torch.compiler backend的config参数,具体定义如下:
1 2 3 4 5 6 7 8 9 10 11 12 |
class CompilerConfig(): """Set CompilerConfig configuration""" def __init__(self): self.debug = _DebugConfig() self.dump_config = _DataDumpConfig() self.export = _ExportConfig() self.inference_config = _InferenceConfig() self.ge_config = _GEConfig() self.fusion_config = _FusionConfig() self.experimental_config = _ExperimentalConfig() super(CompilerConfig, self).__init__() |
关于CompilerConfig类成员的详细介绍参见表1。
成员名 |
功能说明 |
---|---|
debug |
配置debug调试类功能,配置形式为config.debug.xxx,包括如下功能: |
dump_config |
配置图模式下数据dump功能,配置形式为config.dump_config.xxx,具体参见图模式下数据dump功能。 |
export |
配置离线导图相关功能,配置形式为config.export.xxx,具体介绍参见dynamo导图功能。 |
inference_config |
配置推理场景相关功能,配置形式为config.inference_config.xxx,如动态shape图分档执行功能。 |
ge_config |
配置GE图相关功能,配置形式为config.ge_config.xxx,包括如下功能: |
fusion_config |
配置图融合相关功能,配置形式为config.fusion_config.xxx,具体参见算子融合功能。 |
experimental_config |
配置各种实验性质的功能,配置形式为experimental_config.xxx,包括如下功能: |
父主题: 基本类和接口