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

oops, forgot that debug

Browse Source
This commit is contained in:
George Hotz
2023-02-22 06:58:27 -08:00
parent c5e2126d49
commit 104c3c5e73
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_speed_v_torch.py
View File

@@ -7,10 +7,10 @@ import time
import numpy as np
np.set_printoptions(linewidth=160)
from functools import partial
from tinygrad.ops import GlobalCounters, DEBUG
from tinygrad.ops import GlobalCounters
from tinygrad.tensor import Tensor
from tinygrad.nn import Conv2d
from tinygrad.helpers import colored, getenv
from tinygrad.helpers import colored, getenv, DEBUG
from extra.jit import TinyJit
METAL = getenv("METAL")
try: