From 7045109592a7f45bfe1447b4aabe527bfaef8a95 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?=
 <58388001+SzymonOzog@users.noreply.github.com>
Date: Wed, 16 Aug 2023 07:17:08 +0200
Subject: [PATCH] Deprecated includes removed

---
 tinygrad/runtime/ops_triton.py | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/tinygrad/runtime/ops_triton.py b/tinygrad/runtime/ops_triton.py
index d4a2399191..1ab2df4e62 100644
--- a/tinygrad/runtime/ops_triton.py
+++ b/tinygrad/runtime/ops_triton.py
@@ -1,25 +1,17 @@
 from __future__ import annotations
 import hashlib
-from weakref import WeakValueDictionary
-from torch import float32
-import numpy as np
-import pycuda.autoprimaryctx # type: ignore # noqa: F401
 import pycuda.driver as cuda # type: ignore
 
-import triton # type: ignore # noqa: F401
-import triton.language as tl  # type: ignore # noqa: F401
 from triton.compiler import compile as triton_compile
 
-from typing import Any, Union, Tuple, Optional, Dict, List, Final, Callable
-from tinygrad.ops import UnaryOps, BinaryOps, ReduceOps, LazyOp, Op, GlobalCounters, Compiled
-from tinygrad.shape.shapetracker import ShapeTracker
+from typing import Any, Tuple, Dict, List, Final, Callable
+from tinygrad.ops import UnaryOps, BinaryOps, ReduceOps, Op, Compiled
 from tinygrad.helpers import prod, DEBUG, dtypes, ImageDType
-from tinygrad.runtime.ops_gpu import RawBufferCopyInOut
 from tinygrad.runtime.ops_cuda import RawCUDABuffer
-from tinygrad.runtime.lib import RawBuffer
-from tinygrad.codegen.linearizer import LinearizerOptions, UOp, UOps, LocalBuffer
+from tinygrad.codegen.linearizer import LinearizerOptions, UOp, UOps
 from tinygrad.shape.symbolic import NumNode
 
+
 class TritonProgram:
 
   def __init__(self, name:str, prg:str, binary:bool=False):