From faa66e0a613666d773ab87afbb65b8a47833feb9 Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Wed, 25 Feb 2026 21:30:18 +0300 Subject: [PATCH] mi350 hive_reset am repro (#15014) --- test/external/external_test_hive_reset.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/test/external/external_test_hive_reset.py b/test/external/external_test_hive_reset.py index 3399fad8b4..3b0edc4117 100755 --- a/test/external/external_test_hive_reset.py +++ b/test/external/external_test_hive_reset.py @@ -2,8 +2,9 @@ import subprocess, sys from tinygrad.helpers import getenv -LOOPS = getenv("LOOPS", 10) +LOOPS = getenv("LOOPS", 50) BROKEN = getenv("BROKEN", 0) +ONLY_RESET = getenv("ONLY_RESET", 0) BROKEN_KERNEL_SCRIPT = """ from tinygrad.device import Device @@ -36,7 +37,7 @@ for i in range(LOOPS): print(f"=== Running broken kernel ({i+1}/{LOOPS}) ===") ret = subprocess.run([sys.executable, "-c", BROKEN_KERNEL_SCRIPT]) print(f"=== broken kernel exited with code {ret.returncode} ===") - - print(f"=== Running test_tiny.py ({i+1}/{LOOPS}) ===") - ret = subprocess.run([sys.executable, "test/test_tiny.py", "TestTiny.test_plus"]) - print(f"=== test_tiny.py exited with code {ret.returncode} ===") + elif not ONLY_RESET: + print(f"=== Running test_tiny.py ({i+1}/{LOOPS}) ===") + ret = subprocess.run([sys.executable, "test/test_tiny.py", "TestTiny.test_plus"]) + print(f"=== test_tiny.py exited with code {ret.returncode} ===")