Skip to content

Commit 026fe0b

Browse files
authored
Add support for bits16 in ETDump
Differential Revision: D65552835 Pull Request resolved: #6697
1 parent 17ad8d3 commit 026fe0b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

devtools/inspector/_inspector_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@ def get_scalar_type_size(scalar_type: ScalarType) -> Tuple[torch.dtype, int]:
112112
ScalarType.BYTE: (torch.uint8, 1),
113113
ScalarType.CHAR: (torch.int8, 1),
114114
ScalarType.BOOL: (torch.bool, 1),
115+
ScalarType.BITS16: (torch.uint16, 2),
115116
ScalarType.SHORT: (torch.int16, 2),
116117
ScalarType.HALF: (torch.float16, 2),
117118
ScalarType.INT: (torch.int, 4),

0 commit comments

Comments
 (0)