From 82cc8d34bfec01fe13d6acba4edf436f87a3e822 Mon Sep 17 00:00:00 2001
From: Tarun Karuturi <tkaruturi@meta.com>
Date: Fri, 3 Jan 2025 10:57:35 -0800
Subject: [PATCH] Add uint16 to supported dtypes in exir/dialects

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
---
 exir/dialects/edge/dtype/supported.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/exir/dialects/edge/dtype/supported.py b/exir/dialects/edge/dtype/supported.py
index f0772ab4891..89c269de1d0 100644
--- 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 = {