andrey/cblas-src
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fmt fix & CI
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Andrey Tkachenko 2022-03-06 00:00:52 +04:00
parent 3e7ce44242
commit ddfe99dcab
3 changed files with 80 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
.drone.yml Normal file
View File

@ -0,0 +1,13 @@
kind: pipeline
name: default
steps:
- name: build
image: hub.aidev.ru/rust-blas
commands:
- cargo build --verbose --all
- name: fmt-check
image: hub.aidev.ru/rust-blas
commands:
- cargo fmt --all -- --check

61
build.rs
View File

@ -49,30 +49,49 @@ fn main() {
extract(cblas_name, &out_path).unwrap();
}
Command::new("gfortran")
.current_dir(cblas_dir.join("src"))
.arg("-O3")
.arg("-c").arg(cblas_dir.join("src/cdotcsub.f"))
.arg("-c").arg(cblas_dir.join("src/cdotusub.f"))
.arg("-c").arg(cblas_dir.join("src/dasumsub.f"))
.arg("-c").arg(cblas_dir.join("src/ddotsub.f"))
.arg("-c").arg(cblas_dir.join("src/dnrm2sub.f"))
.arg("-c").arg(cblas_dir.join("src/dsdotsub.f"))
.arg("-c").arg(cblas_dir.join("src/dzasumsub.f"))
.arg("-c").arg(cblas_dir.join("src/dznrm2sub.f"))
.arg("-c").arg(cblas_dir.join("src/icamaxsub.f"))
.arg("-c").arg(cblas_dir.join("src/idamaxsub.f"))
.arg("-c").arg(cblas_dir.join("src/isamaxsub.f"))
.arg("-c").arg(cblas_dir.join("src/izamaxsub.f"))
.arg("-c").arg(cblas_dir.join("src/sasumsub.f"))
.arg("-c").arg(cblas_dir.join("src/scasumsub.f"))
.arg("-c").arg(cblas_dir.join("src/scnrm2sub.f"))
.arg("-c").arg(cblas_dir.join("src/sdotsub.f"))
.arg("-c").arg(cblas_dir.join("src/sdsdotsub.f"))
.arg("-c").arg(cblas_dir.join("src/snrm2sub.f"))
.arg("-c").arg(cblas_dir.join("src/zdotcsub.f"))
.arg("-c").arg(cblas_dir.join("src/zdotusub.f"))
.arg("-c")
.arg(cblas_dir.join("src/cdotcsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/cdotusub.f"))
.arg("-c")
.arg(cblas_dir.join("src/dasumsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/ddotsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/dnrm2sub.f"))
.arg("-c")
.arg(cblas_dir.join("src/dsdotsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/dzasumsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/dznrm2sub.f"))
.arg("-c")
.arg(cblas_dir.join("src/icamaxsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/idamaxsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/isamaxsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/izamaxsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/sasumsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/scasumsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/scnrm2sub.f"))
.arg("-c")
.arg(cblas_dir.join("src/sdotsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/sdsdotsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/snrm2sub.f"))
.arg("-c")
.arg(cblas_dir.join("src/zdotcsub.f"))
.arg("-c")
.arg(cblas_dir.join("src/zdotusub.f"))
.status()
.expect("fortran failed");

50
src/lib.rs
View File

@ -1,4 +1,3 @@
#[cfg(test)]
mod tests {
extern crate openblas_src;
@ -8,43 +7,36 @@ mod tests {
#[test]
fn test1() {
let (m, n, k) = (2, 4, 3);
let a = vec![
1.0, 4.0,
2.0, 5.0,
3.0, 6.0,
];
let a = vec![1.0, 4.0, 2.0, 5.0, 3.0, 6.0];
let b = vec![
1.0, 5.0, 9.0,
2.0, 6.0, 10.0,
3.0, 7.0, 11.0,
4.0, 8.0, 12.0,
];
let mut c = vec![
2.0, 7.0,
6.0, 2.0,
0.0, 7.0,
4.0, 2.0,
1.0, 5.0, 9.0, 2.0, 6.0, 10.0, 3.0, 7.0, 11.0, 4.0, 8.0, 12.0,
];
let mut c = vec![2.0, 7.0, 6.0, 2.0, 0.0, 7.0, 4.0, 2.0];
let sdot_res = unsafe {
ddot(6, &a, 1, &a, 1)
};
let sdot_res = unsafe { ddot(6, &a, 1, &a, 1) };
unsafe {
dgemm(Layout::ColumnMajor, Transpose::None, Transpose::None,
m, n, k, 1.0, &a, m, &b, k, 1.0, &mut c, m);
dgemm(
Layout::ColumnMajor,
Transpose::None,
Transpose::None,
m,
n,
k,
1.0,
&a,
m,
&b,
k,
1.0,
&mut c,
m,
);
}
assert_eq!(sdot_res, 91.0);
assert!(
c == vec![
40.0, 90.0,
50.0, 100.0,
50.0, 120.0,
60.0, 130.0,
]
);
assert!(c == vec![40.0, 90.0, 50.0, 100.0, 50.0, 120.0, 60.0, 130.0,]);
}
}