tvm_tirx.tirx.script.call_kernel

Contents

tvm_tirx.tirx.script.call_kernel#

tvm_tirx.tirx.script.call_kernel(kernel, launch_args, *args, **kwargs)#

Call an external kernel.

Parameters:

  • kernel (Any) – The external kernel to call.

  • launch_args (List[Union[int, tirx.Expr, List[Union[int, tirx.Expr]]]]) – The launch arguments. A list of integers for grid size, block size, and shared memory size. The actual requirements depend on the kernel.

  • args (List[tirx.Expr]) – The arguments to pass to the kernel.

  • kwargs (Dict[str, Any]) – Additional keyword arguments to pass to the kernel or compilation.

Contents