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

viz: skip plaintext disassemble for cfg (#13821)

Browse Source
This commit is contained in:
qazal
2025-12-24 23:16:59 +09:00
committed by GitHub
parent cb07c5d0e8
commit e3a646dce3
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tinygrad/viz/serve.py
View File

@@ -428,12 +428,12 @@ def get_render(i:int, j:int, fmt:str) -> dict:
if fmt == "code": return {"src":data.src, "lang":"cpp"}
if fmt == "asm":
compiler = Device[data.device].compiler
disasm_str = get_stdout(lambda: compiler.disassemble(compiler.compile(data.src)))
ret:dict = {"src":disasm_str}
ret:dict = {"metadata":[]}
if data.device.startswith("AMD"):
with soft_err(lambda err: ret.update(err)):
metadata = amd_readelf(lib:=compiler.compile(data.src))
ret = {"data":amdgpu_cfg(lib, getattr(compiler, "arch")), "metadata":[metadata]}
ret["data"] = amdgpu_cfg(lib:=compiler.compile(data.src), getattr(compiler, "arch"))
with soft_err(lambda err: ret["metadata"].append(err)): ret["metadata"].append(amd_readelf(lib))
else: ret["src"] = get_stdout(lambda: compiler.disassemble(compiler.compile(data.src)))
return ret
if fmt == "all-pmc":
durations, pmc = data