test fixups / speedups / var_vals refactor (#13812)

* no PYTHONPATH + llm server port 0

* llm tok speedup

* refactor var_vals
This commit is contained in:
George Hotz
2025-12-23 12:05:59 -05:00
committed by GitHub
parent 8dcba2e2cc
commit 6439a515be
5 changed files with 21 additions and 23 deletions

0
test/device/__init__.py Normal file
View File

View File

@@ -8,7 +8,7 @@ from hypothesis import given, settings, strategies as strat
import numpy as np
import torch
settings.register_profile("my_profile", max_examples=200, deadline=None, derandomize=getenv("DERANDOMIZE_CI", False))
settings.register_profile("my_profile", max_examples=50, deadline=None, derandomize=getenv("DERANDOMIZE_CI", False))
settings.load_profile("my_profile")
core_dtypes = list(DTYPES_DICT.values())

View File

@@ -27,8 +27,8 @@ class TestLLMServer(unittest.TestCase):
from tinygrad.apps.llm import Handler
from tinygrad.helpers import TCPServerWithReuse
cls.port = 11435
cls.server = TCPServerWithReuse(('127.0.0.1', cls.port), Handler)
cls.server = TCPServerWithReuse(('127.0.0.1', 0), Handler)
cls.port = cls.server.server_address[1]
cls.server_thread = threading.Thread(target=cls.server.serve_forever, daemon=True)
cls.server_thread.start()
time.sleep(0.1)