github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: external_test_speed_theoretical works on 24GB

Browse Source
This commit is contained in:
George Hotz
2024-07-26 18:41:52 -07:00
parent b75d1e8793
commit 2fde2d2914
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/external/external_test_speed_theoretical.py vendored
View File

@@ -13,10 +13,10 @@ def _test(tcount, fxn, szmax):
ts = [(x+1).realize() for x in ts]
Device.default.synchronize()
st = time.perf_counter()
out = jfxn(*ts)
out_nbytes = jfxn(*ts).nbytes()
Device.default.synchronize()
tms.append(time.perf_counter() - st)
gbs = (out.nbytes()+sum(x.nbytes() for x in ts))*1e-9/min(tms)
gbs = (out_nbytes+sum(x.nbytes() for x in ts))*1e-9/min(tms)
print(f"{ts[0].nbytes()/(1024*1024):10.0f} MB, {min(tms)*1e3:6.2f} ms GB/s {gbs:<10.2f} {str(ts[0].shape):20s}")
allgbs.append(gbs)
return max(allgbs)