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

remove cpu prereqs (#3410)

Browse Source
This commit is contained in:
George Hotz
2024-02-15 13:45:06 +01:00
committed by GitHub
parent a40df14fef
commit 93eceef727
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/unit/test_disk_tensor.py
View File

@@ -168,7 +168,7 @@ class TestSafetensors(unittest.TestCase):
def helper_test_disk_tensor(fn, data, np_fxn, tinygrad_fxn=None):
if tinygrad_fxn is None: tinygrad_fxn = np_fxn
pathlib.Path(temp(fn)).unlink(missing_ok=True)
tinygrad_tensor = Tensor(data, device="CPU").to(f"disk:{temp(fn)}")
tinygrad_tensor = Tensor(data, device="CLANG").to(f"disk:{temp(fn)}")
numpy_arr = np.array(data)
tinygrad_fxn(tinygrad_tensor)
np_fxn(numpy_arr)
@@ -182,7 +182,7 @@ class TestDiskTensor(unittest.TestCase):
def test_write_ones(self):
pathlib.Path(temp("dt2")).unlink(missing_ok=True)
out = Tensor.ones(10, 10, device="CPU").contiguous()
out = Tensor.ones(10, 10, device="CLANG").contiguous()
outdisk = out.to(f"disk:{temp('dt2')}")
print(outdisk)
outdisk.realize()