diff --git a/accel/llvm/ops_llvm.py b/accel/llvm/ops_llvm.py index 2b6c0f2a9f..4908b4bc1a 100644 --- a/accel/llvm/ops_llvm.py +++ b/accel/llvm/ops_llvm.py @@ -9,7 +9,7 @@ from tinygrad.ops import LazyOp import ctypes import numpy as np from ctypes import CFUNCTYPE -from tinygrad.ops import DEBUG, UnaryOps, BinaryOps, ReduceOps, MovementOps, get_buffers, get_lazyops, ExplicitExecAST, get_lazyop_shape +from tinygrad.ops import DEBUG, UnaryOps, BinaryOps, ReduceOps, get_buffers, get_lazyops, ExplicitExecAST, get_lazyop_shape from llvmlite import ir # type: ignore import llvmlite.binding as llvm # type: ignore diff --git a/tinygrad/llops/ops_gpu.py b/tinygrad/llops/ops_gpu.py index d48def3775..1349f18b40 100644 --- a/tinygrad/llops/ops_gpu.py +++ b/tinygrad/llops/ops_gpu.py @@ -5,7 +5,7 @@ import pyopencl as cl # type: ignore from collections import defaultdict from typing import List, Tuple, Optional, Dict, Union, Set from tinygrad.helpers import prod, ConvArgs, dedup -from tinygrad.ops import DEBUG, ProcessingOps, UnaryOps, BinaryOps, ReduceOps, MovementOps, LazyOp, get_buffers, get_lazyops, Op, get_lazyop_shape, ExplicitExecAST +from tinygrad.ops import DEBUG, ProcessingOps, UnaryOps, BinaryOps, ReduceOps, LazyOp, get_buffers, get_lazyops, Op, get_lazyop_shape, ExplicitExecAST from tinygrad.shapetracker import ShapeTracker CLCACHE = int(os.getenv("CLCACHE", "1"))