From 1e02b4cae18620202d9aa623bec50976afc11d0e Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Mon, 3 Jun 2024 18:21:36 -0400
Subject: [PATCH] default skip all exception in beam (#4822)

added a flag `BEAM_STRICT_MODE` to catch compile error or other exceptions on demand
---
 tinygrad/engine/search.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tinygrad/engine/search.py b/tinygrad/engine/search.py
index 35fe90655f..b18cd6b61c 100644
--- a/tinygrad/engine/search.py
+++ b/tinygrad/engine/search.py
@@ -70,6 +70,9 @@ def _try_compile_linearized_w_idx(x:Tuple[int,Linearizer], compiler:Compiler) ->
     ret = None
   except TimeoutException:
     ret = None
+  except Exception as e:
+    if getenv("BEAM_STRICT_MODE"): raise e
+    ret = None
   finally:
     signal.alarm(0)
   return x[0], ret