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

ane refactor, fix examples

Browse Source
This commit is contained in:
George Hotz
2020-12-18 09:19:06 -08:00
parent fbcd1912cf
commit e5f2055776
4 changed files with 27 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/benchmark.py
View File

@@ -7,7 +7,7 @@ import time
ba = Tensor(np.random.normal(size=(0x4000,)))
for dev in ["CPU", "GPU", "ANE"]:
if dev == "GPU":
baa = ba.cuda()
baa = ba.gpu()
elif dev == "ANE":
baa = ba.ane()
else: