mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-15 09:33:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

ugh, that too

Browse Source
This commit is contained in:
George Hotz
2022-11-07 21:21:35 -08:00
parent 537a9eb414
commit 9884be2ad5
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_speed_v_torch.py
View File

@@ -6,11 +6,12 @@ import time
import numpy as np
from tinygrad.tensor import Tensor
from tinygrad.nn import Conv2d
from tinygrad.llops.ops_gpu import CL
try:
from termcolor import colored
from tinygrad.llops.ops_gpu import CL
except ImportError:
colored = None
CL = None
IN_CHANS = [int(x) for x in os.getenv("IN_CHANS", "4,16,64").split(",")]
@@ -34,7 +35,7 @@ def test_speed(f1, *args):
del ret
st = time.monotonic()
ret = f1(*args)
if ret.device in ["GPU", "OPENCL"]:
if CL is not None and ret.device in ["GPU", "OPENCL"]:
CL.cl_queue.finish()
et = (time.monotonic() - st) * 1000
ets.append(et)