github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-03 19:25:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

move optimize_local_size (#2221)

Browse Source 
* move optimize_local_size

* interpret_ast
This commit is contained in:
George Hotz
2023-11-05 21:00:52 -08:00
committed by GitHub
parent c60c3b467a
commit 2f7aab3d13
16 changed files with 67 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/external_copy_benchmark.py vendored
View File

@@ -2,7 +2,7 @@ import unittest
from tinygrad.helpers import prod
from tinygrad.ops import Device
from tinygrad.tensor import Tensor
from tinygrad.ops import GlobalCounters
from tinygrad.helpers import GlobalCounters
from tinygrad.jit import CacheCollector
class TestCopy(unittest.TestCase):

2
test/external/external_test_allocator_on_models.py vendored
View File

@@ -3,7 +3,7 @@ import unittest, gc
import numpy as np
from tinygrad.tensor import Tensor
from tinygrad.nn.state import get_state_dict
from tinygrad.ops import GlobalCounters
from tinygrad.helpers import GlobalCounters
from tinygrad.runtime.lib import RawBuffer, LRUAllocator
from tinygrad.helpers import dtypes, prod
from tinygrad.ops import Device

2
test/external/external_test_opt.py vendored
View File

@@ -13,7 +13,7 @@ from tinygrad.tensor import Tensor, Device
from tinygrad import nn
from tinygrad.helpers import getenv
from tinygrad.nn import optim
from tinygrad.ops import GlobalCounters, MovementOps, ReduceOps
from tinygrad.helpers import GlobalCounters
from tinygrad.lazy import PUSH_PERMUTES
from tinygrad.jit import CacheCollector