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,
|
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,9 +538,8 @@ where
|
||||||
csc as isize,
|
csc as isize,
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue