warp.overload#

warp.overload(kernel, arg_types=None)[source]#

Overload a generic kernel with the given argument types.

Can be called directly or used as a function decorator.

Parameters:
  • kernel (Kernel | Callable) – The generic kernel to be instantiated with concrete types.

  • arg_types (dict[str, Any] | list[Any] | None) – A list of concrete argument types for the kernel or a dictionary specifying generic argument names as keys and concrete types as variables.

Return type:

Kernel