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