Fixing no-std

coreylowman · Oct 25, 2023 · d36c60a · d36c60a
1 parent 9a2564a
commit d36c60a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/dfdx-core/src/tensor/error.rs b/dfdx-core/src/tensor/error.rs
@@ -1,13 +1,14 @@
+/// 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 {
     /// Device is out of memory
     OutOfMemory,
     /// Not enough elements were provided when creating a tensor
     WrongNumElements,
-
-    UnusedTensors(Vec<crate::tensor::UniqueId>),
-
+    /// Some tensors were unused by an optimizer in a graph.
+    UnusedTensors(std::vec::Vec<crate::tensor::UniqueId>),
     #[cfg(feature = "cuda")]
     CublasError(cudarc::cublas::result::CublasError),
     #[cfg(feature = "cuda")]