AOE工具提供了自定义知识库合并的功能,支持将不同路径下的自定义知识库进行合并,并将其中优于内置知识库中的调优策略存入目标路径中。
仅支持相同昇腾AI处理器型号下的OPAT自定义知识库合并。
拷贝知识库的时候,需要拷贝整个自定义知识库路径(即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
安装CANN软件后,使用CANN运行用户编译、运行时,需要以CANN运行用户登录环境,执行source ${install_path}/latest/bin/setenv.bash命令设置环境变量。其中${install_path}为CANN软件的安装目录。
python3.7 ${install_path}/latest/compiler/python/site-packages/schedule_search.egg/schedule_search/msoptune.py --merge ${src_dir1}:${src_dir2} ${dst_dir}
后续模型编译时若想使用合并后的知识库,需要通过环境变量“TUNE_BANK_PATH”指定合并后的自定义知识库路径。环境变量详细说明请参见配置环境变量。
msoptune.py --merge脚本包含如下功能: