tvm_tirx.tirx.script.ir_builder.external_kernel.BaseKernel

class tvm_tirx.tirx.script.ir_builder.external_kernel.BaseKernel

Bases: object

Base class for external kernels.

compile_to_device_module(launch_args, *args, **kwargs)

Compile the kernel to a device module.

Return type:

tuple[str, Module, list[Any]]