huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Get the cpu tests to run.

Browse Source
This commit is contained in:
laurent
2023-06-28 14:32:02 +01:00
parent 14449ff80c
commit caafef6cc1
5 changed files with 6 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-core/src/tensor.rs
View File

@ -317,6 +317,7 @@ impl Tensor {
unary_op!(sqrt, Sqrt);
unary_op!(gelu, Gelu);
unary_op!(relu, Relu);
pub fn to_scalar<S: crate::WithDType>(&self) -> Result<S> {
if self.rank() != 0 {
return Err(Error::UnexpectedNumberOfDims {