singe-cublas 0.1.0-alpha.5

Safe Rust wrappers for the NVIDIA cuBLAS dense linear algebra library (with cuBLASLt).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use std::ffi::c_void;

use singe_cublas_sys as sys;

use crate::{error::Result, try_ffi, utility::to_usize};

pub fn set_attribute(
    mut call: impl FnMut(*const c_void, core::ffi::c_ulong) -> sys::cublasStatus_t,
    value: *const c_void,
    size: usize,
) -> Result<()> {
    try_ffi!(call(value, size as _))?;
    Ok(())
}

pub fn read_attribute(
    mut call: impl FnMut(
        *mut c_void,
        core::ffi::c_ulong,
        *mut core::ffi::c_ulong,
    ) -> sys::cublasStatus_t,
    value: *mut c_void,
    size: usize,
    name: &str,
) -> Result<usize> {
    let mut written = 0;
    try_ffi!(call(value, size as _, &raw mut written,))?;
    to_usize(written, name)
}