Fix for None dtype

Xilinx · May 23, 2024 · 975edaa · 975edaa
1 parent 8a9196b
commit 975edaa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/brevitas/core/quant/float.py b/src/brevitas/core/quant/float.py
@@ -60,6 +60,8 @@ def __init__(
         self.float_clamp_impl = float_clamp_impl
 
         # To avoid log(0), we add small a small value based on the used dtype
+        if dtype is None:
+            dtype = torch.get_default_dtype()
         self.eps = torch.finfo(dtype).tiny
 
     @brevitas.jit.script_method