github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

move time_linearizer to extra.optimization.helpers [pr] (#9048)

Browse Source 
no longer used in tinygrad
This commit is contained in:
chenyu
2025-02-12 15:49:58 -05:00
committed by GitHub
parent c15486cf39
commit f4f56d7c15
12 changed files with 41 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/external/external_benchmark_hcopt.py vendored
View File

@@ -1,7 +1,7 @@
import random
from tinygrad.helpers import getenv
from tinygrad.engine.search import time_linearizer, beam_search, bufs_from_lin
from extra.optimization.helpers import load_worlds, ast_str_to_lin
from tinygrad.engine.search import beam_search, bufs_from_lin
from extra.optimization.helpers import load_worlds, ast_str_to_lin, time_linearizer
def optimize_kernel(k):
# TODO: update this

4
test/external/speed_beam_v_hcopt.py vendored
View File

@@ -1,7 +1,7 @@
from tinygrad import Device
from tinygrad.helpers import getenv, DEBUG, BEAM
from tinygrad.engine.search import beam_search, time_linearizer, bufs_from_lin
from extra.optimization.helpers import load_worlds, ast_str_to_lin
from tinygrad.engine.search import beam_search, bufs_from_lin
from extra.optimization.helpers import load_worlds, ast_str_to_lin, time_linearizer
if __name__ == "__main__":
filter_reduce = bool(getenv("FILTER_REDUCE"))

3
test/external/verify_kernel.py vendored
View File

@@ -1,10 +1,9 @@
import argparse
from collections import defaultdict
from extra.optimization.helpers import kern_str_to_lin
from extra.optimization.helpers import kern_str_to_lin, time_linearizer
from test.external.fuzz_linearizer import compare_linearizer
from tinygrad.helpers import colored
from tinygrad.codegen.kernel import Kernel
from tinygrad.engine.search import time_linearizer
# Use this with the LOGKERNS options to verify that all executed kernels are valid and evaluate to the same ground truth results

4
test/test_linearizer_overflows.py
View File

@@ -4,8 +4,8 @@ from test.helpers import ast_const
from tinygrad import dtypes, Device
from tinygrad.helpers import CI
from tinygrad.codegen.kernel import Kernel
from tinygrad.engine.search import Opt, OptOps
from tinygrad.engine.search import time_linearizer, bufs_from_lin
from tinygrad.engine.search import Opt, OptOps, bufs_from_lin
from extra.optimization.helpers import time_linearizer
# stuff needed to unpack a kernel
from tinygrad.ops import UOp, Ops

3
test/test_search.py
View File

@@ -4,7 +4,7 @@ from test.helpers import ast_const
from tinygrad.codegen.kernel import Opt, OptOps
from tinygrad.codegen.kernel import Kernel
from tinygrad.ops import UOp, Ops
from tinygrad.engine.search import time_linearizer, bufs_from_lin, actions, beam_search
from tinygrad.engine.search import bufs_from_lin, actions, beam_search
from tinygrad.device import Device, Buffer
from tinygrad.tensor import Tensor
from tinygrad.dtype import dtypes
@@ -12,6 +12,7 @@ from tinygrad.helpers import Context, GlobalCounters
from tinygrad.engine.realize import capturing
from tinygrad.shape.shapetracker import ShapeTracker
from tinygrad.shape.view import View
from extra.optimization.helpers import time_linearizer
class TestTimeLinearizer(unittest.TestCase):
@unittest.skipIf(Device.DEFAULT == "WEBGPU", "WebGPU timestamps are low precision, tm is 0")