coaster_blas/frameworks/cuda/
mod.rs

1//! Provides BLAS for a CUDA backend.
2#![allow(missing_docs)]
3use crate::cublas;
4use crate::plugin::*;
5use crate::transpose::Transpose;
6use coaster::backend::Backend;
7use coaster::frameworks::cuda::Cuda;
8use coaster::plugin::Error as PluginError;
9use coaster::tensor::{ITensorDesc, SharedTensor};
10
11#[macro_use]
12pub mod helper;
13
14impl Asum<f32> for Backend<Cuda> {
15    iblas_asum_for_cuda!(f32);
16}
17
18impl Axpy<f32> for Backend<Cuda> {
19    iblas_axpy_for_cuda!(f32);
20}
21
22impl Copy<f32> for Backend<Cuda> {
23    iblas_copy_for_cuda!(f32);
24}
25
26impl Dot<f32> for Backend<Cuda> {
27    iblas_dot_for_cuda!(f32);
28}
29
30impl Nrm2<f32> for Backend<Cuda> {
31    iblas_nrm2_for_cuda!(f32);
32}
33
34impl Scal<f32> for Backend<Cuda> {
35    iblas_scal_for_cuda!(f32);
36}
37
38impl Swap<f32> for Backend<Cuda> {
39    iblas_swap_for_cuda!(f32);
40}
41
42impl Gbmv<f32> for Backend<Cuda> {
43    iblas_gbmv_for_cuda!(f32);
44}
45
46impl Gemm<f32> for Backend<Cuda> {
47    iblas_gemm_for_cuda!(f32);
48}
coaster_blas/frameworks/cuda/mod.rs

coaster_blas/frameworks/cuda/
mod.rs
