算子在线编译选项
功能简介
模型编译时,支持用户在线编译和使用编译好的算子二进制文件。用户根据实际情况自行设置。
使用方法
该功能通过torchair.get_npu_backend中compiler_config参数配置,配置示例如下,参数说明参见表1。
1 2 3 4 5 |
import torch_npu, torchair config = torchair.CompilerConfig() # 算子在线编译选项配置 config.experimental_config.jit_compile = "auto" npu_backend = torchair.get_npu_backend(compiler_config=config) |
父主题: 更多功能