From b3dd885ffb6d9faab57a9a51c4456c97b010a4e0 Mon Sep 17 00:00:00 2001 From: chenyu Date: Thu, 16 May 2024 14:21:22 -0400 Subject: [PATCH] cleanup double import from tinygrad.device in tensor.py (#4620) --- tinygrad/tensor.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index c768b57879cf..71589842a51c 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -7,13 +7,12 @@ import numpy as np from tinygrad.dtype import DType, dtypes, ImageDType, ConstType, least_upper_float, least_upper_dtype, sum_acc_dtype -from tinygrad.helpers import argfix, make_pair, flatten, prod, all_int, round_up, merge_dicts, fully_flatten, argsort, IMAGE, DEBUG, WINO, THREEFRY -from tinygrad.helpers import getenv +from tinygrad.helpers import argfix, make_pair, flatten, prod, all_int, round_up, merge_dicts, fully_flatten, argsort, getenv +from tinygrad.helpers import IMAGE, DEBUG, WINO, THREEFRY from tinygrad.lazy import LazyBuffer from tinygrad.multi import MultiLazyBuffer from tinygrad.ops import LoadOps -from tinygrad.device import Buffer, BufferOptions -from tinygrad.device import Device +from tinygrad.device import Device, Buffer, BufferOptions from tinygrad.shape.symbolic import sint, Variable, MulNode, Node from tinygrad.engine.realize import run_schedule from tinygrad.engine.schedule import ScheduleItem, create_schedule_with_vars, memory_planner