Kernel Call

In warp, launching a kernel is tedious and lost all type-hint of the kernel function:

@warp.kernel
def func(a: int, b: float): ...

warp.launch(func, shape, inputs = (1, 1.0))

warp_hint enable you directly call the kernel function with shape provieded in a subsequent call:

from warp_hint import kernel 

@kernel
def func(a: int, b: float): ...

func(1, 1.0)(shape) # type-hint is avaliable