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
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

//#[cfg_attr(link(name = "BLAS", kind = "framework"))]
//#[cfg_attr(target_os = "windows", link(name = "BLAS"))] ??

use crate::devices::cpu::{Order, Transpose};

#[link(name = "blas")]
//#[cfg_attr(target_os = "macos", link(name = "blas"))]
//#[cfg_attr(target_os = "linux", link(name = "openblas"))]
extern "C" {

    pub(crate) fn cblas_sgemm(
        order: Order,
        trans_a: Transpose,
        trans_b: Transpose,
        m: usize,
        n: usize,
        k: usize,
        alpha: f32,
        a: *const f32,
        lda: usize,
        b: *const f32,
        ldb: usize,
        beta: f32,
        c: *mut f32,
        ldc: usize,
    );

    pub(crate) fn cblas_dgemm(
        order: Order,
        trans_a: Transpose,
        trans_b: Transpose,
        m: usize,
        n: usize,
        k: usize,
        alpha: f64,
        a: *const f64,
        lda: usize,
        b: *const f64,
        ldb: usize,
        beta: f64,
        c: *mut f64,
        ldc: usize,
    );
}