Skip to content

Commit 4b1323d

Browse files
add support for uint16 in inspector
Differential Revision: D66305565 Pull Request resolved: #7011
1 parent 2d51e63 commit 4b1323d

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
@@ -113,6 +113,7 @@ def get_scalar_type_size(scalar_type: ScalarType) -> Tuple[torch.dtype, int]:
113113
ScalarType.CHAR: (torch.int8, 1),
114114
ScalarType.BOOL: (torch.bool, 1),
115115
ScalarType.BITS16: (torch.uint16, 2),
116+
ScalarType.UINT16: (torch.uint16, 2),
116117
ScalarType.SHORT: (torch.int16, 2),
117118
ScalarType.HALF: (torch.float16, 2),
118119
ScalarType.INT: (torch.int, 4),

0 commit comments

Comments
 (0)