文档
注册
评分
提单
论坛
小AI

ENABLE_ACLNN

功能描述

该环境变量在后续版本会废弃。

图编译时,通过设置该环境变量,决定是否在图执行时调用算子注册的host执行函数实现host执行逻辑以及kernel下发。取值如下:

  • true:图执行时调用算子注册的host执行函数实现host执行逻辑以及kernel下发。
  • false:图执行时仍旧使用GE执行器做算子的kernel下发。

算子支持情况如下:

  • 若算子仅支持host侧单算子执行调用接口(信息库中aclnnSupport取值配置为aclnn_only),配置该环境变量不生效,始终调用host执行函数实现host执行逻辑以及kernel下发。
  • 若算子不支持host侧单算子执行调用接口(信息库中未配置aclnnSupport字段),则保持原有使用GE执行器做算子的kernel下发。
  • 若算子两种方式都支持,则只有环境变量取值为true+信息库标记aclnnSupport(信息库中aclnnSupport取值配置为support_aclnn)+动态shape(图的shape包含-1或者-2)三个条件都满足,才调用host执行函数实现host执行逻辑以及kernel下发;否则仍旧使用GE执行器做算子的kernel下发。

配置示例

export ENABLE_ACLNN=true

使用约束

支持的型号

Atlas 训练系列产品

Atlas 推理系列产品(Ascend 310P处理器)

Atlas A2训练系列产品/Atlas 800I A2推理产品

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词