View page source

warp.Kernel#

class warp.Kernel(
func,
key=None,
module=None,
options=None,
code_transformers=None,
source=None,
)[source]#

Warp kernel object, typically created by decorating a Python function with @warp.kernel.

Kernels can be launched on CPU or GPU devices using warp.launch().

Methods

add_overload(arg_types)

get_mangled_name()

get_overload(arg_types)

infer_argument_types(args)

grid_stride: bool | None#
infer_argument_types(args)[source]#
add_overload(arg_types)[source]#
get_overload(arg_types)[source]#
get_mangled_name()[source]#
Return type:

str