根据输入的模板库Kernel信息,生成Kernel下发代码。
gen_file = mskpp.Launcher(config).code_gen()
参数名 |
输入/输出 |
说明 |
|---|---|---|
gen_file |
输入 |
指定生成Kernel侧下发代码的文件路径。 数据类型:str。 可选参数,默认值为_gen_launch.cpp。 |
生成代码的文件路径。
config = mskpp.KernelInvokeConfig(kernel_file, kernel_name) gen_file = mskpp.Launcher(config).code_gen()
class KernelInvokeConfig:
...
A configuration descriptor for a possible kernel developed based on an Act example
...
def __init__(self, kernel_src_file : str, kernel_name : str):
pass
# 用户仅能传KernelInvokeConfig类型
class Launcher:
def __init__(self, config: KernelInvokeConfig):
...
a class that generates launch source code for a kernel
Args:
config (KernelInvokeConfig): A configuration descriptor for a kernel
...