Skip to content

Commit 9008017

Browse files
authored
Add uint16 to supported dtypes in exir/dialects
Differential Revision: D67803963 Pull Request resolved: #7495
1 parent e66cdaf commit 9008017

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

exir/dialects/edge/dtype/supported.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
torch.float16: "Half",
2626
torch.float: "Float",
2727
torch.double: "Double",
28+
torch.uint16: "UInt16",
2829
}
2930

3031
regular_tensor_str_to_dtypes = {

0 commit comments

Comments
 (0)