Serialize storage_offset in sym int.

zhxchen17 · facebook-github-bot · commit 4f11d93e773d · 2024-03-05T13:12:09.000-08:00
Summary: as title.

Differential Revision: D54553809
diff --git a/exir/serde/export_serialize.py b/exir/serde/export_serialize.py
@@ -220,7 +220,7 @@ def serialize_tensor_meta(t: torch.Tensor) -> TensorMeta:
         requires_grad=t.requires_grad,
         device=Device(type=t.device.type, index=t.device.index),
         strides=[serialize_sym_int(s) for s in t.stride()],
-        storage_offset=0,
+        storage_offset=serialize_sym_int(0),
         layout=_TORCH_TO_SERIALIZE_LAYOUT[t.layout],
     )
 

Original file line number	Diff line number	Diff line change
`@@ -220,7 +220,7 @@ def serialize_tensor_meta(t: torch.Tensor) -> TensorMeta:`
`220`	`220`	`requires_grad=t.requires_grad,`
`221`	`221`	`device=Device(type=t.device.type, index=t.device.index),`
`222`	`222`	`strides=[serialize_sym_int(s) for s in t.stride()],`
`223`		`- storage_offset=0,`
	`223`	`+ storage_offset=serialize_sym_int(0),`
`224`	`224`	`layout=_TORCH_TO_SERIALIZE_LAYOUT[t.layout],`
`225`	`225`	`)`
`226`	`226`