Add uint16 to supported dtypes in exir/dialects

Differential Revision: D67803963 Pull Request resolved: #7495
pytorch · Jan 3, 2025 · 9008017 · 9008017
1 parent e66cdaf
commit 9008017
Showing 1 changed file with 1 addition and 0 deletions.
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 = {