From af357b5dc817152157f14fde48c13ead4e3fa70c Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Thu, 31 Jul 2025 04:22:08 +0800 Subject: [PATCH] disable TRACK_MATCH_STATS in BEAM workers [pr] (#11437) --- tinygrad/opt/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/opt/search.py b/tinygrad/opt/search.py index 2a798f16c1..41875ab043 100644 --- a/tinygrad/opt/search.py +++ b/tinygrad/opt/search.py @@ -83,7 +83,7 @@ def _try_compile_linearized_w_idx(x:tuple[int,Kernel], compiler:Compiler) -> tup # workers should not open devices and should ignore ctrl c and should not launch VIZ def _init_worker(): - Context(ALLOW_DEVICE_USAGE=0, VIZ=0).__enter__() + Context(ALLOW_DEVICE_USAGE=0, VIZ=0, TRACK_MATCH_STATS=0).__enter__() signal.signal(signal.SIGINT, signal.SIG_IGN) def _ensure_buffer_alloc(bufs:list[Buffer]) -> list[Buffer]: return [buf.ensure_allocated() if buf is not None else buf for buf in bufs]