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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

//! Provides BLAS for a CUDA backend.
#![allow(missing_docs)]
use crate::cublas;
use crate::plugin::*;
use crate::transpose::Transpose;
use coaster::backend::Backend;
use coaster::frameworks::cuda::Cuda;
use coaster::plugin::Error as PluginError;
use coaster::tensor::{ITensorDesc, SharedTensor};

#[macro_use]
pub mod helper;

impl Asum<f32> for Backend<Cuda> {
    iblas_asum_for_cuda!(f32);
}

impl Axpy<f32> for Backend<Cuda> {
    iblas_axpy_for_cuda!(f32);
}

impl Copy<f32> for Backend<Cuda> {
    iblas_copy_for_cuda!(f32);
}

impl Dot<f32> for Backend<Cuda> {
    iblas_dot_for_cuda!(f32);
}

impl Nrm2<f32> for Backend<Cuda> {
    iblas_nrm2_for_cuda!(f32);
}

impl Scal<f32> for Backend<Cuda> {
    iblas_scal_for_cuda!(f32);
}

impl Swap<f32> for Backend<Cuda> {
    iblas_swap_for_cuda!(f32);
}

impl Gbmv<f32> for Backend<Cuda> {
    iblas_gbmv_for_cuda!(f32);
}

impl Gemm<f32> for Backend<Cuda> {
    iblas_gemm_for_cuda!(f32);
}