add back final check?

tinygrad · May 25, 2024 · 92f990c · 92f990c
1 parent 7535ea8
commit 92f990c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tinygrad/function.py b/tinygrad/function.py
@@ -828,9 +828,9 @@ def forward(self, x: LazyBuffer) -> LazyBuffer:
             TernaryOps.WHERE, computed, computed.const(float("inf"))
         )
         computed = isnotnan.e(TernaryOps.WHERE, computed, x.const(float("nan")))
-        # computed = initial_x.e(BinaryOps.CMPLT, ninf_t).e(
-        #     TernaryOps.WHERE, computed.const(0), computed
-        # )
+        computed = initial_x.e(BinaryOps.CMPLT, ninf_t).e(
+            TernaryOps.WHERE, computed.const(0), computed
+        )
 
         # print("COMPUTED: ")
         # print(__import__('tinygrad').Tensor(computed).numpy()[0])