fix lazy value comparison bug

inisis · Oct 19, 2024 · d2e09c3 · d2e09c3
1 parent b3f9742
commit d2e09c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxslim/third_party/onnx_graphsurgeon/ir/tensor.py b/onnxslim/third_party/onnx_graphsurgeon/ir/tensor.py
@@ -315,7 +315,7 @@ def __eq__(self, other):
         if not isinstance(other, LazyValues):
             return False
 
-        tensor_match = self.tensor.raw_data == other.tensor.raw_data
+        tensor_match = self.tensor == other.tensor
         shape_match = self.shape == other.shape
         dtype_match = self.dtype == other.dtype