github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-17 01:51:40 -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_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