已关闭
[Feature]: 静态编译支持按进程隔离 #2634
rich创建于  7月7日关闭于  7月9日
rich
rich成员
7月7日 创建

提交提案之前,请先检索仓库内是否已有相同的提案,如已有请在同一提案中进行讨论。

💻 需求背景、当前现状、期望实现的功能内容、具体的设计方案、以及测试方案

1.1 现状

当前静态编译的 .run 包安装到 CANN 全局路径 $ASCEND_HOME_PATH/../*/opp/static_kernel/ai_core/<pkg>/,同一节点上所有进程共享同一安装路径。这带来以下问题:

  • 进程间冲突:多个进程同时安装不同的 static kernel 会互相覆盖
  • 清理风险:异常退出时残留的 kernel 需要手动执行 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 指定当前进程使用的静态算子路径
  • 旧版 torch_npu(无新接口)自动回退到原有行为

替代方案

补充说明

欢迎加入社区,感谢您对社区的贡献 🎉!

likedislike
richrich成员
7月7日 添加了label:feature
此处折叠了19条事件消息 查看更多
ascend-robotascend-robot成员
7月13日 添加了label:resolved