mulimoen
/
SummationByParts
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Neon supports fma

This commit is contained in:
Magnus Ulimoen 2021-04-18 23:20:08 +02:00
parent 2ee55de8c8
commit 091076a9ad
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/constmatrix/src/lib.rs
View File

@ -132,7 +132,7 @@ impl<const M: usize, const P: usize> Matrix<Float, M, P> {
}; };
for k in 1..N { for k in 1..N {
cfg_if::cfg_if!( cfg_if::cfg_if!(
if #[cfg(target_feature="fma")] { if #[cfg(any(target_feature="fma", target_feature="neon"))] {
t = Float::mul_add(lhs[(i, k)], rhs[(k, j)], t); t = Float::mul_add(lhs[(i, k)], rhs[(k, j)], t);
} else { } else {
t = t + lhs[(i, k)]*rhs[(k, j)]; t = t + lhs[(i, k)]*rhs[(k, j)];