From eac58eaabaa9c27444873a8bb4878dc0c97cd79f Mon Sep 17 00:00:00 2001
From: George Hotz <72895+geohot@users.noreply.github.com>
Date: Thu, 17 Oct 2024 00:21:04 +0800
Subject: [PATCH] no SIGALRM on windows [pr] (#7104)

---
 tinygrad/engine/search.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/tinygrad/engine/search.py b/tinygrad/engine/search.py
index e727da0f62e39..1d2dea66f25ab 100644
--- a/tinygrad/engine/search.py
+++ b/tinygrad/engine/search.py
@@ -55,9 +55,10 @@ class TimeoutException(Exception): pass
 def timeout_handler(signum, frame): raise TimeoutException()
 
 def _try_compile_linearized_w_idx(x:Tuple[int,Kernel], compiler:Compiler) -> Tuple[int, Optional[Tuple[Program, bytes, float]]]:
-  signal.signal(signal.SIGALRM, timeout_handler)
-  # set timeout
-  signal.alarm(getenv("BEAM_TIMEOUT_SEC", 10))
+  if hasattr(signal, "SIGALRM"):
+    signal.signal(signal.SIGALRM, timeout_handler)
+    # set timeout
+    signal.alarm(getenv("BEAM_TIMEOUT_SEC", 10))
   try:
     p = x[1].to_program(name_override="test")
     assert p.uops is not None, "uop list wasn't generated?"
@@ -75,7 +76,7 @@ def _try_compile_linearized_w_idx(x:Tuple[int,Kernel], compiler:Compiler) -> Tup
     if getenv("BEAM_STRICT_MODE"): raise e
     ret = None
   finally:
-    signal.alarm(0)
+    if hasattr(signal, "SIGALRM"): signal.alarm(0)
   return x[0], ret
 
 # workers should ignore ctrl c