Update src/onnx_ir/passes/common/common_subexpression_elimination.py

Co-authored-by: Justin Chu <[email protected]>
Signed-off-by: Ti-Tai Wang <[email protected]>

Author: titaiwangms

src/onnx_ir/passes/common/common_subexpression_elimination.py

@@ -81,11 +81,12 @@ def _eliminate_common_subexpression(self, graph: ir.Graph, modified: bool) -> bo
                     # to ensure they are hashable.
                     value = tuple(value)
                 elif v.type is ir.AttributeType.TENSOR:
-                    np_value = value.numpy()
-                    if np_value.nbytes > self.size_limit:
+                    if value.size > self.size_limit:
                         # If the tensor is larger than the size limit, we skip it.
                         large_tensor = True
                         break
+                    np_value = value.numpy()
+
                     value = (np_value.shape, str(np_value.dtype), np_value.tobytes())
                 attributes[k] = value