tvm_tirx.tirx.script.ir_builder.external_kernel

Contents

tvm_tirx.tirx.script.ir_builder.external_kernel#

External kernel integration fro TIR

Classes#

BaseKernel()

Base class for external kernels.

SourceKernel(source_code)

A kernel from source code.

Functions#

call_kernel(kernel, launch_args, *args, **kwargs)

Call an external kernel.