diff --git a/src/brevitas/quant_tensor/int_quant_tensor.py b/src/brevitas/quant_tensor/int_quant_tensor.py index 072eccff4..6e9f1053a 100644 --- a/src/brevitas/quant_tensor/int_quant_tensor.py +++ b/src/brevitas/quant_tensor/int_quant_tensor.py @@ -106,9 +106,9 @@ def int(self, float_datatype=False): else: return int_value.type(torch.float32) else: - if self.bit_width <= 8. and self.signed.item(): + if self.bit_width <= 8. and self.signed(): return int_value.to(torch.int8) - elif self.bit_width <= 8. and not self.signed.item(): + elif self.bit_width <= 8. and not self.signed(): return int_value.to(torch.uint8) else: return int_value.to(torch.int32)