tvm_tirx.tirx.script.ir_builder.external_kernel.SourceKernel

tvm_tirx.tirx.script.ir_builder.external_kernel.SourceKernel#

class tvm_tirx.tirx.script.ir_builder.external_kernel.SourceKernel(source_code)#

Bases: BaseKernel

A kernel from source code.

Parameters:

source_code (str)

compile_to_device_module(grid, *args, **kwargs)#

Compile the kernel to a device module.

Parameters:

  • grid (list[list[int | Expr]])

  • args (list[Any])

  • kwargs (dict[str, Any])

Return type:

tuple[str, Module, list[Any]]