Skip to content

Commit

Permalink
Disable torch.uint8 quantization (#3295)
Browse files Browse the repository at this point in the history
Signed-off-by: Sai Chaitanya Gajula <[email protected]>
  • Loading branch information
quic-gsaichai authored Sep 3, 2024
1 parent aa99065 commit 19aae4f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion TrainingExtensions/torch/src/python/aimet_torch/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@
logger = AimetLogger.get_area_logger(AimetLogger.LogAreas.Utils)

dtypes_to_ignore_for_quantization = (int, float, bool, str, tuple, type(None))
torch_dtypes_to_ignore_for_quantization = [torch.int, torch.int8, torch.int16, torch.int32, torch.int64, torch.bool]
torch_dtypes_to_ignore_for_quantization = [torch.int, torch.int8, torch.int16, torch.int32, torch.int64, torch.bool, torch.uint8]
allowed_output_types = (torch.Tensor, *dtypes_to_ignore_for_quantization)
DROPOUT_TYPES = (torch.nn.Dropout, torch.nn.Dropout2d, torch.nn.Dropout3d)

Expand Down

0 comments on commit 19aae4f

Please sign in to comment.