diff --git a/devtools/inspector/_inspector_utils.py b/devtools/inspector/_inspector_utils.py index f7126443035..c2e92f09148 100644 --- a/devtools/inspector/_inspector_utils.py +++ b/devtools/inspector/_inspector_utils.py @@ -112,6 +112,7 @@ def get_scalar_type_size(scalar_type: ScalarType) -> Tuple[torch.dtype, int]: ScalarType.BYTE: (torch.uint8, 1), ScalarType.CHAR: (torch.int8, 1), ScalarType.BOOL: (torch.bool, 1), + ScalarType.BITS16: (torch.uint16, 2), ScalarType.SHORT: (torch.int16, 2), ScalarType.HALF: (torch.float16, 2), ScalarType.INT: (torch.int, 4),