昇腾社区首页
中文
注册

compile

功能说明

编译kernel下发代码,返回一个可执行的kernel对象。

函数原型

kernel = compile(build_script, gen_file)

参数说明

参数名

输入/输出

说明

build_script

输入

用于模板库Kernel编译的脚本。

数据类型:str。

必选参数。

gen_file

输入

由code_gen接口生成的Kernel下发代码文件路径,一般直接使用code_gen接口返回值。

数据类型:str。

必选参数。

output_bin_path

输入

指定编译生成的可执行文件路径。

数据类型:str。

可选参数,默认值:_gen_module.so。

use_cache

输入

开启后不执行编译,加载output_bin_path所指定的文件。

数据类型:bool。

可选参数,默认值:False。

返回值说明

可运行的Kernel对象。

Kernel对象类型为CompiledKernel,支持如下方式调用kernel:kernel[blockdim](arg1, arg2, ...)。

调用示例

kernel = compile(build_script, gen_file)