From d36c60afcd92384c4b204cac1034058547d0dbce Mon Sep 17 00:00:00 2001 From: Corey Lowman Date: Wed, 25 Oct 2023 12:39:56 -0400 Subject: [PATCH] Fixing no-std --- dfdx-core/src/tensor/error.rs | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/dfdx-core/src/tensor/error.rs b/dfdx-core/src/tensor/error.rs index 76c31bda..f6b43c32 100644 --- a/dfdx-core/src/tensor/error.rs +++ b/dfdx-core/src/tensor/error.rs @@ -1,3 +1,5 @@ +/// Represents a number of different errors that can occur from creating tensors +/// or launching tensor operations. This encompasses both Cpu and CUDA errors. #[non_exhaustive] #[derive(Debug)] pub enum Error { @@ -5,9 +7,8 @@ pub enum Error { OutOfMemory, /// Not enough elements were provided when creating a tensor WrongNumElements, - - UnusedTensors(Vec), - + /// Some tensors were unused by an optimizer in a graph. + UnusedTensors(std::vec::Vec), #[cfg(feature = "cuda")] CublasError(cudarc::cublas::result::CublasError), #[cfg(feature = "cuda")]