sysblas 0.1.1

Minimal CBLAS FFI bindings to system BLAS
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

extern crate sysblas;

// NB: These tests mostly exist to check that linking succeeded.

#[test]
fn test_sasum() {
  let x: Vec<f32> = vec![1.0, -2.0, 3.0];
  let r = unsafe { sysblas::cblas_sasum(
      x.len() as _,
      x.as_ptr(), 1) };
  assert_eq!(6.0, r);
}

#[test]
fn test_saxpy() {
  let x: Vec<f32> = vec![1.0, 2.0, 3.0];
  let mut y: Vec<f32> = vec![3.0, 2.0, 1.0];
  unsafe { sysblas::cblas_saxpy(
      x.len() as _,
      1.0,
      x.as_ptr(), 1,
      y.as_mut_ptr(), 1) };
  assert_eq!(&[4.0, 4.0, 4.0], y.as_slice());
}

#[test]
fn test_scopy() {
  let x: Vec<f32> = vec![1.0, 2.0, 3.0];
  let mut y: Vec<f32> = vec![3.0, 2.0, 1.0];
  unsafe { sysblas::cblas_scopy(
      x.len() as _,
      x.as_ptr(), 1,
      y.as_mut_ptr(), 1) };
  assert_eq!(&[1.0, 2.0, 3.0], y.as_slice());
}

#[test]
fn test_sdot() {
  let x: Vec<f32> = vec![1.0, 2.0, 3.0];
  let y: Vec<f32> = vec![3.0, 2.0, 1.0];
  let r = unsafe { sysblas::cblas_sdot(
      x.len() as _,
      x.as_ptr(), 1,
      y.as_ptr(), 1) };
  assert_eq!(10.0, r);
}

#[test]
fn test_snrm2() {
  let x: Vec<f32> = vec![3.0, 4.0, 3.0, 4.0, 3.0, 4.0, 3.0, 4.0];
  let r = unsafe { sysblas::cblas_snrm2(
      x.len() as _,
      x.as_ptr(), 1) };
  assert_eq!(10.0, r);
}

#[test]
fn test_sscal() {
  let mut x: Vec<f32> = vec![1.0, 2.0, 3.0];
  unsafe { sysblas::cblas_sscal(
      x.len() as _,
      2.0,
      x.as_mut_ptr(), 1) };
  assert_eq!(&[2.0, 4.0, 6.0], x.as_slice());
}

#[test]
fn test_sgemv() {
  let a: Vec<f32> = vec![0.0, 0.0, 1.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0];
  let x: Vec<f32> = vec![1.0, 2.0, 3.0];
  let mut y: Vec<f32> = vec![0.0, 0.0, 0.0];
  unsafe { sysblas::cblas_sgemv(
      sysblas::CblasColMajor,
      sysblas::CblasNoTrans,
      3, 3,
      1.0,
      a.as_ptr(), 3,
      x.as_ptr(), 1,
      0.0,
      y.as_mut_ptr(), 1) };
  assert_eq!(&[2.0, 3.0, 1.0], y.as_slice());
}

#[test]
fn test_sgemm() {
  let a: Vec<f32> = vec![0.0, 0.0, 1.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0];
  let b: Vec<f32> = vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0];
  let mut c: Vec<f32> = vec![0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0];
  unsafe { sysblas::cblas_sgemm(
      sysblas::CblasColMajor,
      sysblas::CblasNoTrans,
      sysblas::CblasNoTrans,
      3, 3, 3,
      1.0,
      a.as_ptr(), 3,
      b.as_ptr(), 3,
      0.0,
      c.as_mut_ptr(), 3) };
  assert_eq!(&[2.0, 3.0, 1.0, 5.0, 6.0, 4.0, 8.0, 9.0, 7.0], c.as_slice());
}