allow upcast again

tinygrad · May 24, 2024 · 44d7fbd · 44d7fbd
1 parent 29ea9fc
commit 44d7fbd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py
@@ -1132,8 +1132,8 @@ def contiguous(self): return F.Contiguous.apply(self)
   def contiguous_backward(self): return F.ContiguousBackward.apply(self)
   def log(self): return F.Log.apply(self.cast(least_upper_float(self.dtype)))
   def log2(self): return self.log()/math.log(2)
-  # def exp(self): return F.Exp.apply(self.cast(least_upper_float(self.dtype)))
-  def exp(self): return F.Exp.apply(self)
+  def exp(self): return F.Exp.apply(self.cast(least_upper_float(self.dtype)))
+  # def exp(self): return F.Exp.apply(self)
   def exp2(self): return F.Exp.apply(self*math.log(2))
   def relu(self): return F.Relu.apply(self)
   def sigmoid(self): return F.Sigmoid.apply(self.cast(least_upper_float(self.dtype)))