Skip to content

Commit 36c8342

Browse files
tarun292facebook-github-bot
authored andcommitted
Add uint16 to supported dtypes and regenerate edge.yaml (#7687)
Summary: Adding `uint16` support and regenerating `edge.yaml` as we now support `uint16` in the ExecuTorch stack. Differential Revision: D68241997
1 parent a727b55 commit 36c8342

File tree

2 files changed

+831
-481
lines changed

2 files changed

+831
-481
lines changed

exir/dialects/edge/dtype/runner.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,9 @@
1717

1818
class DtypeRunner:
1919
def __init__(self):
20-
self.tensor_dtypes = list(common_dtype.all_types_and(torch.bool, torch.half))
20+
self.tensor_dtypes = list(
21+
common_dtype.all_types_and(torch.bool, torch.half, torch.uint16)
22+
)
2123
self.scalar_dtypes = [torch.bool, torch.int, torch.float]
2224

2325
@staticmethod

0 commit comments

Comments
 (0)