github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-26 23:38:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

VIZ show rendering errors (#6807)

Browse Source 
* VIZ show rendering errors

* show the entire traceback
This commit is contained in:
qazal
2024-09-30 09:35:36 +08:00
committed by GitHub
parent 2ed94e447f
commit 7099af4450
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
viz/serve.py
View File

@@ -1,7 +1,7 @@
#!/usr/bin/env python3
from __future__ import annotations
from typing import Dict, List, Optional, Tuple
import pickle, os, sys, time, threading, webbrowser, json, difflib, contextlib, re
import pickle, os, sys, time, threading, webbrowser, json, difflib, contextlib, re, traceback
from dataclasses import dataclass, asdict
from urllib.parse import parse_qs, urlparse
from http.server import HTTPServer, BaseHTTPRequestHandler
@@ -96,10 +96,12 @@ def load_kernels(contexts:List[TrackedRewriteContext]) -> List[KernelRet]:
kernel_name = ""
code = ""
for ctx in contexts:
if ctx.loc[0].split("/")[-1] == "schedule.py":
si_ctx = ScheduleItemContext(bufs=tuple(x.arg for x in ctx.sink.sparents if x.op is UOps.BUFFER))
with Context(TRACK_MATCH_STATS=0): kernel_name, code = (prg:=get_runner(Device.DEFAULT, full_ast_rewrite(ctx.sink, si_ctx)).p).name, prg.src
elif ctx.kernel_name is not None: kernel_name, code = ctx.kernel_name, ""
try:
if ctx.loc[0].split("/")[-1] == "schedule.py":
si_ctx = ScheduleItemContext(bufs=tuple(x.arg for x in ctx.sink.sparents if x.op is UOps.BUFFER))
with Context(TRACK_MATCH_STATS=0): kernel_name, code = (prg:=get_runner(Device.DEFAULT, full_ast_rewrite(ctx.sink, si_ctx)).p).name, prg.src
elif ctx.kernel_name is not None: kernel_name, code = ctx.kernel_name, ""
except Exception: kernel_name, code = "RENDERING_ERROR", traceback.format_exc()
if ret.get(k:=to_function_name(kernel_name)) is None: ret[k] = KernelRet(k, code, [])
ret[k].ctxs.append(ctx)
return list(ret.values())