huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add some missing backtraces. (#1193)

Browse Source
This commit is contained in:
Laurent Mazare
2023-10-27 06:09:11 +01:00
committed by GitHub
parent 70d06ab4b0
commit 9b1158b315
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1186,14 +1186,16 @@ impl Tensor {
op: "scatter-add (self, src)", op: "scatter-add (self, src)",
lhs: self.shape().clone(), lhs: self.shape().clone(),
rhs: source.shape().clone(), rhs: source.shape().clone(),
})? }
.bt())?
} }
if indexes.dims() != source.dims() { if indexes.dims() != source.dims() {
Err(Error::ShapeMismatchBinaryOp { Err(Error::ShapeMismatchBinaryOp {
op: "scatter-add (indexes, src)", op: "scatter-add (indexes, src)",
lhs: indexes.shape().clone(), lhs: indexes.shape().clone(),
rhs: source.shape().clone(), rhs: source.shape().clone(),
})? }
.bt())?
} }
let storage = self.storage().scatter_add( let storage = self.storage().scatter_add(
self.layout(), self.layout(),
@ -1265,7 +1267,8 @@ impl Tensor {
op: "slice-scatter (self, src)", op: "slice-scatter (self, src)",
lhs: self.shape().clone(), lhs: self.shape().clone(),
rhs: src.shape().clone(), rhs: src.shape().clone(),
})? }
.bt())?
} }
let mut storage = self.device().zeros(self.shape(), self.dtype())?; let mut storage = self.device().zeros(self.shape(), self.dtype())?;
self.storage() self.storage()
@ -1299,7 +1302,8 @@ impl Tensor {
op: "index-add (self, source)", op: "index-add (self, source)",
lhs: self.shape().clone(), lhs: self.shape().clone(),
rhs: source.shape().clone(), rhs: source.shape().clone(),
})? }
.bt())?
} }
// The number of element in indexes must match the dimension on which the add is // The number of element in indexes must match the dimension on which the add is
// performed on the source tensor (and the index values from `indexes` are taken from // performed on the source tensor (and the index values from `indexes` are taken from
@ -1310,7 +1314,8 @@ impl Tensor {
op: "index-add (ids, source))", op: "index-add (ids, source))",
lhs: indexes.shape().clone(), lhs: indexes.shape().clone(),
rhs: source.shape().clone(), rhs: source.shape().clone(),
})? }
.bt())?
} }
let storage = self.storage().index_add( let storage = self.storage().index_add(
self.layout(), self.layout(),
@ -1358,7 +1363,8 @@ impl Tensor {
op: "gather", op: "gather",
lhs: self.shape().clone(), lhs: self.shape().clone(),
rhs: indexes.shape().clone(), rhs: indexes.shape().clone(),
})? }
.bt())?
} }
let storage = let storage =
self.storage() self.storage()