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

View File

@ -132,7 +132,7 @@ impl<const M: usize, const P: usize> Matrix<Float, M, P> {
};
for k in 1..N {
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);
} else {
t = t + lhs[(i, k)]*rhs[(k, j)];