mirror of
https://github.com/huggingface/candle.git
synced 2025-06-19 19:58:35 +00:00
Add the cat operator (without the storage implementation for now).
This commit is contained in:
11
src/error.rs
11
src/error.rs
@ -9,6 +9,9 @@ pub enum Error {
|
||||
#[error("{op} only supports contiguous tensors")]
|
||||
RequiresContiguous { op: &'static str },
|
||||
|
||||
#[error("{op} expects at least one tensor")]
|
||||
OpRequiresAtLeastOneTensor { op: &'static str },
|
||||
|
||||
#[error("the candle crate has not been built with cuda support")]
|
||||
NotCompiledWithCudaSupport,
|
||||
|
||||
@ -24,6 +27,14 @@ pub enum Error {
|
||||
op: &'static str,
|
||||
},
|
||||
|
||||
#[error("shape mismatch in cat for dim {dim}, shape for arg 1: {first_shape:?} shape for arg {n}: {nth_shape:?}")]
|
||||
ShapeMismatchCat {
|
||||
dim: usize,
|
||||
first_shape: Shape,
|
||||
n: usize,
|
||||
nth_shape: Shape,
|
||||
},
|
||||
|
||||
#[error("device mismatch in {op}, lhs: {lhs:?}, rhs: {rhs:?}")]
|
||||
DeviceMismatchBinaryOp {
|
||||
lhs: DeviceLocation,
|
||||
|
Reference in New Issue
Block a user