下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

compile_ops_adv_npu

函数功能

ops_adv算子工程场景下NPU侧编译接口。根据用户配置信息,进行算子源码编译,生成kernel.o文件。

函数原型

def compile_ops_adv_npu(self, repo_path: str, tiling_key: int, options: CompileNpuOptions, repo_type: str = 'ops_adv')

参数说明

参数名

输入/输出

说明

是否必设

repo_path

输入

本地算子源码仓路径,例如ops_adv仓路径/work_code/ops/ops_adv。

tiling_key

输入

Tiling key值。

options

输入

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

repo_type

输入

算子开发场景类型。 缺省默认值为ops_adv。

返回值

函数返回如下三个结果:

  • kernel name。
  • kernel.o绝对路径文件名。
  • 编译信息dict:
    • 关键字'cross_core_sync'值是bool类型,表示是否开启硬同步syncall。
    • 关键字'task_ration'值是list类型,表示核函数运行时的core分配方案。

使用说明

仅支持ops_adv算子工程场景,调用示例参见核函数NPU上板精度验证

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

当前产品无相关内容

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