文档
注册

run_custom_cpu

函数功能

标准自定义工程算子开发场景的CPU调测接口。根据输入的Tiling bin文件,进行算子g++编译,生成CPU侧可执行程序。在CPU上运行该程序,完成算子运行仿真,并自动完成输出数据精度比对。

函数原型

def run_custom_cpu(self, customize_path: str, tiling_info: TilingInfo, cpu_options: CpuOptions)

参数说明

参数名

输入/输出

说明

是否必设

customize_path

输入

自定义算子安装包路径。根据实际部署环境配置。

例如${INSTALL_DIR}/opp/vendors/${vendor_name},其中${INSTALL_DIR}为CANN安装包路径,${vendor_name}为部署自定义算子包时设置的vendor名。

tiling_info

输入

TilingInfo对象,包含Tiling bin文件名、Tiling key、workspace、block_num等信息。

cpu_options

输入

CpuOptions的实例对象,用户根据实际情况自行设置。

返回值

使用说明

仅支持标准自定义工程算子开发场景,调用示例参见核函数CPU孪生调试

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

当前产品无相关内容

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