知识库合并
简介
AOE工具提供了自定义知识库合并的功能,支持将不同路径下的自定义知识库进行合并,并将其中优于内置知识库中的调优策略存入目标路径中。
- 仅支持相同昇腾AI处理器型号下的OPAT自定义知识库合并。
- 名字中携带"runtime_kb"的json文件不支持合并。
操作步骤
- 将不同环境中生成的待合并的知识库拷贝到执行知识库合并操作所在环境,不同环境中生成的知识库请存储在不同的目录下,例如${src_dir1}、${src_dir2}。
拷贝知识库的时候,需要拷贝整个自定义知识库路径(即TUNE_BANK_PATH)及该路径下的内容。例如:自定义知识库路径为/home/aoe/data,${src_dir1}为/home/HwHiAiUser/data,则需要参考如下命令进行拷贝。
cp -r /home/aoe/data /home/HwHiAiUser
- 合并知识库。
- 配置环境变量。
CANN组合包提供进程级环境变量设置脚本,供用户在进程中引用,以自动完成环境变量设置。执行命令参考如下,以下示例均为root或非root用户默认安装路径,请以实际安装路径为准。
# 以root用户安装toolkit包 . /usr/local/Ascend/ascend-toolkit/set_env.sh # 以非root用户安装toolkit包 . ${HOME}/Ascend/ascend-toolkit/set_env.sh
- 执行知识库合并。
python3 $HOME/Ascend/ascend-toolkit/latest/python/site-packages/schedule_search/msoptune.py --merge ${src_dir1}:${src_dir2} ${dst_dir}
- ${src_dirx}:可以配置多个src_dir,不同的src_dir之间使用“:”进行分隔。src_dir为1中指定的存放待合并知识库的${src_dir}。
- ${dst_dir}:配置为合并后自定义知识库的存储路径,可为绝对路径或相对路径,需要为已存在的路径且msoptune.py脚本需要有读、写、可执行权限。
后续模型编译时若想使用合并后的知识库,需要通过环境变量“TUNE_BANK_PATH”指定合并后的自定义知识库路径。环境变量详细说明请参见配置环境变量。
msoptune.py --merge脚本包含如下功能:
- 配置环境变量。
父主题: 附录