gemm: don't return early for scalar types other than f32,f64.
This commit is contained in:
parent
bd3b73f872
commit
6c895ed383
|
@ -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,9 +538,8 @@ where
|
|||
csc as isize,
|
||||
);
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue