Skip to content

Commit 5bd6924

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

File tree

2 files changed

+829
-481
lines changed

2 files changed

+829
-481
lines changed

exir/dialects/edge/dtype/runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
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(common_dtype.all_types_and(torch.bool, torch.half, torch.uint16))
2121
self.scalar_dtypes = [torch.bool, torch.int, torch.float]
2222

2323
@staticmethod

0 commit comments

Comments
 (0)