Only hash tensor desc

tracel-ai · Jan 24, 2025 · 99b5fbe · 99b5fbe
1 parent e0a0317
commit 99b5fbe
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/crates/burn-tensor/src/repr/operation.rs b/crates/burn-tensor/src/repr/operation.rs
@@ -1695,8 +1695,6 @@ impl ModuleOperationDescription {
 impl core::hash::Hash for FromDataOperationDescription {
     fn hash<H: core::hash::Hasher>(&self, state: &mut H) {
         self.out.hash(state);
-
-        self.data.bytes.hash(state);
     }
 }
 

diff --git a/crates/burn-tensor/src/tensor/data.rs b/crates/burn-tensor/src/tensor/data.rs
@@ -36,7 +36,7 @@ pub enum DataError {
 #[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
 pub struct TensorData {
     /// The values of the tensor (as bytes).
-    pub(crate) bytes: Bytes,
+    bytes: Bytes,
 
     /// The shape of the tensor.
     pub shape: Vec<usize>,