diff --git a/tinygrad/viz/serve.py b/tinygrad/viz/serve.py index 73c3222cca..f53f72aa9c 100755 --- a/tinygrad/viz/serve.py +++ b/tinygrad/viz/serve.py @@ -7,7 +7,7 @@ from dataclasses import dataclass, field from urllib.parse import parse_qs, urlparse from http.server import BaseHTTPRequestHandler from typing import Any, TypedDict, TypeVar, Generator, Callable -from tinygrad.helpers import colored, getenv, tqdm, unwrap, word_wrap, TRACEMETA, ProfileEvent, ProfileRangeEvent, TracingKey, ProfilePointEvent, temp +from tinygrad.helpers import colored, getenv, unwrap, word_wrap, TRACEMETA, ProfileEvent, ProfileRangeEvent, TracingKey, ProfilePointEvent, temp from tinygrad.helpers import printable, Context, START_TIME, NO_COLOR, ansistrip from tinygrad.renderer.amd.dsl import Inst from tinygrad.renderer.amd import detect_format @@ -182,7 +182,7 @@ def get_full_rewrite(data:VizData, ctx:TrackedGraphRewrite, depth:int|None=None) next_sink = _reconstruct(data, ctx.sink, depth=depth) yield {"graph":uop_to_json(data, next_sink), "uop":pystr(next_sink), "change":None, "diff":None, "upat":None, "_sink":next_sink} replaces: dict[UOp, UOp] = {} - for u0_num,u1_num,upat_loc,dur in tqdm(ctx.matches, disable=not ctx.matches): + for u0_num,u1_num,upat_loc,dur in ctx.matches: replaces[u0:=_reconstruct(data, u0_num, depth=depth)] = u1 = _reconstruct(data, u1_num, depth=depth) try: new_sink = next_sink.substitute(replaces, walk=ctx.walk, enter_calls=ctx.enter_calls) except RuntimeError as e: new_sink = UOp(Ops.NOOP, arg=str(e))