Skip to content

Commit

Permalink
No item
Browse files Browse the repository at this point in the history
  • Loading branch information
Giuseppe5 committed Dec 27, 2024
1 parent 1297617 commit 4e2c4c5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/brevitas/quant_tensor/int_quant_tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 4e2c4c5

Please sign in to comment.