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

simple runtime args (#2211)

Browse Source 
* simple runtime args

* fix some tests

* fix abstractions and triton

* fix search
This commit is contained in:
George Hotz
2023-11-03 12:31:29 -07:00
committed by GitHub
parent 9ea0448103
commit f17bc16f46
16 changed files with 37 additions and 24 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

@@ -41,7 +41,7 @@ class FakeBuffer(RawBuffer):
def toCPU(self): return np.empty(self.size, dtype=self.dtype.np)
class FakeProgram:
def __init__(self, name:str, prg:str): pass
def __call__(self, global_size, local_size, *bufs, wait=False): pass
def __call__(self, *bufs, global_size, local_size, wait=False): pass
def helper_test_correctness(gen, train):
from tinygrad.runtime.ops_gpu import CL, CLAllocator