当 ASCEND_CUSTOM_OPP_PATH 未设置时,AICPU 自定义算子包(vendor 包)无法被加载,用户必须显式设置环境变量才能使用自定义算子。
ASCEND_CUSTOM_OPP_PATH
PR: https://gitcode.com/cann/opbase/pull/698
GetCustomOppPathFromVendors()
ASCEND_OPP_PATH/vendors/config.ini
load_priority
CustJsonLoadAndParse()
ASCEND_OPP_PATH
cust_aicpu_kernel.json
需求描述
当
ASCEND_CUSTOM_OPP_PATH未设置时,AICPU 自定义算子包(vendor 包)无法被加载,用户必须显式设置环境变量才能使用自定义算子。实现方案
PR: https://gitcode.com/cann/opbase/pull/698
GetCustomOppPathFromVendors():从ASCEND_OPP_PATH/vendors/config.ini读取load_priority字段,推导 vendor 包路径CustJsonLoadAndParse()增加 fallback 分支:CUSTOM_OPP_PATH 未设置或为空串时走 vendors 推导ASCEND_OPP_PATH根目录下直接查找cust_aicpu_kernel.json作为候选涉及文件
关联 PR