From 9b8440ddea272e74d18b52df41aecc4af4774d46 Mon Sep 17 00:00:00 2001 From: Riley Dulin Date: Wed, 6 Nov 2024 11:40:04 -0800 Subject: [PATCH] Add support for bits16 in ETDump Summary: Bits16 is a newer scalar type, and some backends (like Cadence) use it for quantize and dequantize kernels. In case ETDump is turned on for intermediate tensors, it needs to know the size and the corresponding torch dtype. Differential Revision: D65552835 --- devtools/inspector/_inspector_utils.py | 1 + 1 file changed, 1 insertion(+) 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),