Serialize union fields with single entry dict. (#2279)

zhxchen17 · facebook-github-bot · commit 6be02ceec448 · 2024-03-07T11:35:16.000-08:00
Summary: X-link: pytorch/pytorch#121337 Pull Request resolved: #2279 remove "$type" and "$value" fields, instead only serialize as {type: value} for union fields directly. bypass-github-export-checks Reviewed By: tugsbayasgalan Differential Revision: D54600943 fbshipit-source-id: d4bab91541db42eec2d8ead1176db6acf0db8ef3
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`