From c70c112254b1612dbb97bb040bb96fd5923bfa1b Mon Sep 17 00:00:00 2001 From: anu Date: Fri, 9 Jan 2026 01:19:59 -0500 Subject: [PATCH] fix CUDA=1 disassembly (VIZ=1) by stripping null terminator (#14046) * fix ptxas disassembly bug * single ' * move fix to get_bytes * move rstrip --------- Co-authored-by: qazal <77887910+Qazalin@users.noreply.github.com> --- tinygrad/runtime/support/compiler_cuda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/support/compiler_cuda.py b/tinygrad/runtime/support/compiler_cuda.py index 36dc3cdc03..8f71a92551 100644 --- a/tinygrad/runtime/support/compiler_cuda.py +++ b/tinygrad/runtime/support/compiler_cuda.py @@ -35,7 +35,7 @@ def pretty_ptx(s): def cuda_disassemble(lib:bytes, arch:str): try: fn = (pathlib.Path(tempfile.gettempdir()) / f"tinycuda_{hashlib.md5(lib).hexdigest()}").as_posix() - with open(fn, "wb") as f: f.write(lib) + with open(fn, "wb") as f: f.write(lib.rstrip(b'\x00')) subprocess.run(["ptxas", f"-arch={arch}", "-o", fn, fn], check=False, stderr=subprocess.DEVNULL) # optional ptx -> sass step for CUDA=1 print(system(f'nvdisasm {fn}')) except Exception as e: print("Failed to generate SASS", str(e), "Make sure your PATH contains ptxas/nvdisasm binary of compatible version.")