Skip to content

Commit 8cf2680

Browse files
manuelcandalesStrycekSimon
authored andcommitted
Support BFloat16 in exir
Differential Revision: D82129165 Pull Request resolved: pytorch#14164
1 parent be2a082 commit 8cf2680

File tree

5 files changed

+233
-223
lines changed

5 files changed

+233
-223
lines changed

exir/dialects/edge/dtype/supported.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
torch.int32: "Int",
2424
torch.int64: "Long",
2525
torch.float16: "Half",
26+
torch.bfloat16: "BFloat16",
2627
torch.float: "Float",
2728
torch.double: "Double",
2829
torch.uint16: "UInt16",

0 commit comments

Comments
 (0)