cumath 0.2.7

Cuda-based matrix/vector computations
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



mod meta;
mod cuda_core;
mod cuvector;
mod cumatrix;
mod kernel;

pub use cuda_core::{cuda::*, cublas::*, curand::*};
pub use cuvector::*;
pub use cumatrix::*;




use std::fmt::Display;


pub trait CuDataType: Clone + PartialEq + Display {
    #[inline]
    fn zero() -> Self;
}
impl CuDataType for i32 {
    fn zero() -> i32 { 0 }
}
impl CuDataType for f32 {
    fn zero() -> f32 { 0.0 }
}