add support for uint16 in inspector

Differential Revision: D66305565 Pull Request resolved: pytorch#7011
kirklandsign · Nov 22, 2024 · 4b1323d · 4b1323d
1 parent 2d51e63
commit 4b1323d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/devtools/inspector/_inspector_utils.py b/devtools/inspector/_inspector_utils.py
@@ -113,6 +113,7 @@ def get_scalar_type_size(scalar_type: ScalarType) -> Tuple[torch.dtype, int]:
             ScalarType.CHAR: (torch.int8, 1),
             ScalarType.BOOL: (torch.bool, 1),
             ScalarType.BITS16: (torch.uint16, 2),
+            ScalarType.UINT16: (torch.uint16, 2),
             ScalarType.SHORT: (torch.int16, 2),
             ScalarType.HALF: (torch.float16, 2),
             ScalarType.INT: (torch.int, 4),