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对象,类型:CompiledKernel,支持如下方式调用kernel:kernel[blockdim](arg1, arg2, ..., timeout=-1, device_id=0, repeat=1),其中arg1、args2、...是Kernel的入参。
调用示例
kernel = compile(build_script, gen_file) kernel[blockdim](arg1, arg2, ..., device_id=0)
参数名 |
输入/输出 |
说明 |
---|---|---|
device_id |
输入 |
NPU设备ID,设置运行ST测试用例的昇腾AI处理器的ID。 数据类型:int。 若未设置此参数,默认为0。 |
timeout |
输入 |
camodel仿真场景需要默认设置较长超时时间,设置为-1时表示不限制。 数据类型:int。 单位: ms,默认值为300000。 |
repeat |
输入 |
重复运行次数,默认值为1。 数据类型:int。 |
stream |
输入 |
预留参数。 |
kernel_name |
输入 |
预留参数。 |
父主题: 接口列表