Skip to content

Commit 273ff98

Browse files
tarun292YIWENX14
authored andcommitted
Add uint16 to supported dtypes and regenerate edge.yaml
Differential Revision: D68241997 Pull Request resolved: #7687
1 parent b207df7 commit 273ff98

File tree

3 files changed

+832
-481
lines changed

3 files changed

+832
-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)