gemm: don't return early for scalar types other than f32,f64.

This commit is contained in:
sebcrozet 2018-05-20 16:12:24 +02:00 committed by Sébastien Crozet
parent bd3b73f872
commit 6c895ed383
1 changed files with 2 additions and 2 deletions

View File

@ -514,6 +514,7 @@ where
csc as isize, csc as isize,
); );
} }
return;
} else if N::is::<f64>() { } else if N::is::<f64>() {
let (rsa, csa) = a.strides(); let (rsa, csa) = a.strides();
let (rsb, csb) = b.strides(); let (rsb, csb) = b.strides();
@ -537,11 +538,10 @@ where
csc as isize, csc as isize,
); );
} }
}
return; return;
} }
} }
}
for j1 in 0..ncols1 { for j1 in 0..ncols1 {
// FIXME: avoid bound checks. // FIXME: avoid bound checks.