kernelkit.kernel.BaseKernel

kernelkit.kernel.BaseKernel#

class kernelkit.kernel.BaseKernel(cuda_source: str)#

Abstract base class for CUDA kernels

Attributes:
cuda_source

Returns the CUDA source code of the kernel

is_compiled

Methods

__call__(*args, **kwargs)

Call self as a function.

abstract __init__(cuda_source: str)#

Initialize the kernel

Methods

__init__(cuda_source)

Initialize the kernel

Attributes

cuda_source

Returns the CUDA source code of the kernel

is_compiled