Struct KernelRegistry 

pub struct KernelRegistry { /* private fields */ }

Instance-scoped kernel registry.

Implementations

impl KernelRegistry

pub fn new() -> Self

Construct an empty registry.

pub fn register(&self, entry: KernelEntry) -> Result<KernelId, AbiError>

Register a new kernel and return its assigned KernelId.

Returns AbiError::QuotaExceeded if either the per-instance kernel count cap (MAX_KERNELS_PER_INSTANCE) or the aggregate PTX-bytes cap (MAX_TOTAL_PTX_BYTES) would be exceeded.

pub fn lookup( &self, id: KernelId, owner: InstanceId, ) -> Result<KernelHandle, AbiError>

Look up a kernel by id and return an independent handle.

Returns Err(AbiError::InvalidKernel) if the id is unknown or belongs to a different instance. The returned KernelHandle holds an Arc<Module> on CUDA builds, so callers may drop any borrowed dashmap entry before performing the launch — eliminating the UAF that the previous dashmap::Ref-derived raw-pointer scheme had.

pub fn remove(&self, id: KernelId) -> Option<KernelEntry>

Remove a kernel from the registry (caller releases its handle).

On CUDA builds the underlying cust::module::Module is held inside an Arc; dropping the registry entry only decrements the strong count. If a concurrent launch is still holding a clone of the Arc, the module is not actually unloaded until that launch returns — eliminating the UAF window that existed when the registry owned the module by value.

pub fn len(&self) -> usize

Number of currently-registered kernels.

pub fn is_empty(&self) -> bool

True if there are no registered kernels.

pub fn total_ptx_bytes(&self) -> u64

Aggregate PTX bytes currently retained by the registry. Visible for metrics and tests.

Trait Implementations

impl Default for KernelRegistry

fn default() -> Self

Returns the “default value” for a type.