下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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

表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,包括如下功能:

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词