From 104c3c5e73b47a798c2fe2c0eadc261ef8a80515 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 22 Feb 2023 06:58:27 -0800 Subject: [PATCH] oops, forgot that debug --- test/test_speed_v_torch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_speed_v_torch.py b/test/test_speed_v_torch.py index 5742c93040..dd16454434 100644 --- a/test/test_speed_v_torch.py +++ b/test/test_speed_v_torch.py @@ -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: