Skip to main content

Module cublaslt

Module cublaslt 

Source
Expand description

Thin local wrappers over cudarc::cublaslt::sys for the entry points cudarc 0.19.4’s safe layer doesn’t expose. Each helper is unsafe fn if it dereferences raw handles; the safe shape is just “create / set-attribute / destroy” RAII over the opaque handle.

Used by:

Structs§

Preference
RAII handle for a cublasLtMatmulPreference_t. Created via Preference::new, destroyed on Drop.

Functions§

check
Map a cublasStatus_t into an Err(String) for the cuBLASLt status codes we care about. We don’t go through CublasError here because cudarc’s result module doesn’t expose every variant we touch — the string form is sufficient since these errors funnel straight into GpuError::LibraryError.
set_desc_i32_attr
Set an i32-valued attribute on a matmul descriptor (used for EPILOGUE, the cudarc-bindgen cublasLtEpilogue_t is repr(u32) but the cuBLASLt API expects int).
set_desc_pointer_attr
Set a pointer-valued attribute on a matmul descriptor (used for BIAS_POINTER, EPILOGUE_AUX_POINTER, fp8 scale pointers).