Add uint16 to supported dtypes in exir/dialects

tarun292 · facebook-github-bot · commit 82cc8d34bfec · 2025-01-03T10:57:35.000-08:00
Summary: Add `uint16_t` to the supported dtypes here so that the verifier doesn't fail when one of the dtypes for an op is `uint16_t`.

Differential Revision: D67803963
diff --git a/exir/dialects/edge/dtype/supported.py b/exir/dialects/edge/dtype/supported.py
@@ -25,6 +25,7 @@
     torch.float16: "Half",
     torch.float: "Float",
     torch.double: "Double",
+    torch.uint16: "UInt16",
 }
 
 regular_tensor_str_to_dtypes = {

Original file line number	Diff line number	Diff line change
`@@ -25,6 +25,7 @@`
`25`	`25`	`torch.float16: "Half",`
`26`	`26`	`torch.float: "Float",`
`27`	`27`	`torch.double: "Double",`
	`28`	`+ torch.uint16: "UInt16",`
`28`	`29`	`}`
`29`	`30`
`30`	`31`	`regular_tensor_str_to_dtypes = {`