custos 0.7.0

A minimal OpenCL, WGPU, CUDA and host CPU array manipulation engine.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! CUBLAS ffi module

mod error;
mod ffi;

pub use ffi::*;

use self::error::CublasResult;

/// Raw CUBLAS handle
#[derive(Debug)]
pub struct CublasHandle(pub *mut cublasContext);

/// create a CUBLAS handle
pub fn create_handle() -> CublasResult<CublasHandle> {
    let mut handle: CublasHandle = CublasHandle(std::ptr::null_mut());
    unsafe { cublasCreate_v2(&mut handle.0) }.to_result()?;
    Ok(handle)
}