Skip to main content

set_desc_i32_attr

Function set_desc_i32_attr 

Source
pub unsafe fn set_desc_i32_attr(
    desc: cublasLtMatmulDesc_t,
    attr: cublasLtMatmulDescAttributes_t,
    value: i32,
) -> Result<(), String>
Expand description

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).

ยงSafety

desc must be a live cublasLtMatmulDesc_t.