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

_alloc and _free with options (#3934)

Browse Source 
* _alloc has options

* linter

* fix hsa
This commit is contained in:
nimlgen
2024-03-26 19:11:41 +03:00
committed by GitHub
parent 739f47eb0f
commit e2d6f76723
9 changed files with 34 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/external_test_speed_llama.py vendored
View File

@@ -12,7 +12,7 @@ class FakeProgram:
def __call__(self, *bufs, global_size, local_size, vals=(), wait=False): pass
class FakeAllocator(Allocator):
def _alloc(self, sz): return None
def _alloc(self, sz, options): return None
def copyin(self, dest, src:memoryview): pass
class TestLLaMASpeed(unittest.TestCase):