File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -320,10 +320,10 @@ def decoder(self, i):
320
320
else :
321
321
return res
322
322
else :
323
- from jax .numpy import float32
323
+ from jax .numpy import float64
324
324
from jax .core import Tracer
325
325
if isinstance (i , Tracer ):
326
- res = i * float32 (2 )** self .exponent
326
+ res = i * float64 (2 )** self .exponent
327
327
else :
328
328
res = i * 2 ** self .exponent
329
329
@@ -335,8 +335,8 @@ def jdecoder(self, i):
335
335
return self .decoder (i )
336
336
337
337
def encoder (self , i ):
338
- from jax .numpy import float32
339
- res = signed_int_iso_2 (i / (float32 (2 )** self .exponent ), self .size )
338
+ from jax .numpy import float64
339
+ res = signed_int_iso_2 (i / (float64 (2 )** self .exponent ), self .size )
340
340
# if self.signed:
341
341
# res = signed_int_iso(i/2**self.exponent, self.size-1)
342
342
# else:
You can’t perform that action at this time.
0 commit comments