提交提案之前,请先检索仓库内是否已有相同的提案,如已有请在同一提案中进行讨论。
当前静态编译的 .run 包安装到 CANN 全局路径 $ASCEND_HOME_PATH/../*/opp/static_kernel/ai_core/<pkg>/,同一节点上所有进程共享同一安装路径。这带来以下问题:
.run
$ASCEND_HOME_PATH/../*/opp/static_kernel/ai_core/<pkg>/
uninstall.sh
torch_npu 新增 _aclnn_reselect_static_kernel_with_path 接口和 .run 包新增支持 --install-path 参数,实现进程级隔离:
_aclnn_reselect_static_kernel_with_path
--install-path
static_kernel_compile_outputs/install/{timestamp}_{pid}/
reselect_with_path
欢迎加入社区,感谢您对社区的贡献 🎉!
提交提案之前,请先检索仓库内是否已有相同的提案,如已有请在同一提案中进行讨论。
💻 需求背景、当前现状、期望实现的功能内容、具体的设计方案、以及测试方案
1.1 现状
当前静态编译的
.run包安装到 CANN 全局路径$ASCEND_HOME_PATH/../*/opp/static_kernel/ai_core/<pkg>/,同一节点上所有进程共享同一安装路径。这带来以下问题:uninstall.sh清理1.2 目标
torch_npu 新增
_aclnn_reselect_static_kernel_with_path接口和.run包新增支持--install-path参数,实现进程级隔离:static_kernel_compile_outputs/install/{timestamp}_{pid}/目录reselect_with_path指定当前进程使用的静态算子路径替代方案
补充说明
欢迎加入社区,感谢您对社区的贡献 🎉!