已关闭
[AICPU] ASCEND_CUSTOM_OPP_PATH 未设置时无法自动加载自定义算子包 #311
liu-wei创建于  8月13日关闭于  8月13日
liu-wei
liu-wei成员
8月13日 创建

需求描述

当 ASCEND_CUSTOM_OPP_PATH 未设置时,AICPU 自定义算子包(vendor 包)无法被加载,用户必须显式设置环境变量才能使用自定义算子。

实现方案

PR: https://gitcode.com/cann/opbase/pull/698

  1. 新增 GetCustomOppPathFromVendors():从 ASCEND_OPP_PATH/vendors/config.ini 读取 load_priority 字段,推导 vendor 包路径
  2. CustJsonLoadAndParse() 增加 fallback 分支:CUSTOM_OPP_PATH 未设置或为空串时走 vendors 推导
  3. 兜底:ASCEND_OPP_PATH 根目录下直接查找 cust_aicpu_kernel.json 作为候选
  4. 新增 UT 覆盖 5 种场景(直接设置 / 空串 fallback / 未设置 fallback / 多 vendor / 无 config.ini)

涉及文件

  • src/nnopbase/aicpu/task_handler/aicpu_json_load_manager.cpp
  • src/nnopbase/aicpu/task_handler/aicpu_json_load_manager.h
  • tests/nnopbase/ut/CMakeLists.txt
  • tests/nnopbase/ut/individual_op/aicpu_json_load_utest.cpp

关联 PR

  • #698
likedislike
liu-weiliu-wei成员
8月13日 将 liu-wei 设为负责人
CANN-robotCANN-robot成员
8月13日 关闭了 issue
CANN-robotCANN-robot成员
8月13日 添加了label:resolved