简介
概述
TBE提供了基于TVM的DSL算子正确性验证框架,用于在CPU上对DSL算子进行功能调试,验证算子的算法逻辑实现是否正确。
开发者可以使用numpy构造输入golden数据,并将DSL的输出结果与期望结果进行比对,还可以打印中间Tensor,从而帮助开发者进行DSL运算逻辑错误的定位。
您可以从CANN软件安装后文件存储路径的“python/site-packages/tbe/common/testing/testing.py”文件中获取Testing相关接口的定义。
接口使用说明
使用TBE DSL接口前需要参见环境准备完成CANN基础环境变量的配置。
父主题: TBE DSL Testing API