Add uint16 support in exir serde

tarun292 · facebook-github-bot · commit f93b578316c3 · 2024-10-29T15:29:29.000-07:00
Summary: Adding uint16 support in serialization after D64812253.

Differential Revision: D65175529
diff --git a/exir/serde/export_serialize.py b/exir/serde/export_serialize.py
@@ -148,6 +148,7 @@ def _reverse_map(d: Dict[Any, Enum]):
     torch.complex128: ScalarType.COMPLEXDOUBLE,
     torch.bool: ScalarType.BOOL,
     torch.bfloat16: ScalarType.BFLOAT16,
+    torch.uint16: ScalarType.UINT16
 }
 
 
diff --git a/exir/serde/schema.py b/exir/serde/schema.py
@@ -35,7 +35,7 @@ class ScalarType(IntEnum):
     COMPLEXDOUBLE = 11
     BOOL = 12
     BFLOAT16 = 13
-
+    UINT16 = 14
 
 class Layout(IntEnum):
     Unknown = 0

Original file line number	Diff line number	Diff line change
`@@ -148,6 +148,7 @@ def _reverse_map(d: Dict[Any, Enum]):`
`148`	`148`	`torch.complex128: ScalarType.COMPLEXDOUBLE,`
`149`	`149`	`torch.bool: ScalarType.BOOL,`
`150`	`150`	`torch.bfloat16: ScalarType.BFLOAT16,`
	`151`	`+ torch.uint16: ScalarType.UINT16`
`151`	`152`	`}`
`152`	`153`
`153`	`154`