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上板精度验证。
父主题: OpExecutor类