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

bump assembly debug to 7 (#9662)

Browse Source
This commit is contained in:
Ignacio Sica
2025-04-01 11:51:33 +08:00
committed by GitHub
parent ac533e89a2
commit cfad139189
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/engine/realize.py
View File

@@ -41,7 +41,7 @@ class CompiledRunner(Runner):
if DEBUG >= 4: print(p.src)
self.p:ProgramSpec = p
self.lib:bytes = precompiled if precompiled is not None else Device[p.device].compiler.compile_cached(p.src)
if DEBUG >= 6: Device[p.device].compiler.disassemble(self.lib)
if DEBUG >= 7: Device[p.device].compiler.disassemble(self.lib)
self._prg = Device[p.device].runtime(p.function_name, self.lib)
super().__init__(p.name, p.device, p.estimates)

2
tinygrad/runtime/ops_cuda.py
View File

@@ -34,7 +34,7 @@ class CUDAProgram:
def __init__(self, dev:CUDADevice, name:str, lib:bytes, smem:int=0):
self.dev, self.name, self.lib, self.smem = dev, name, lib, smem
if DEBUG >= 5: print("\n".join([f"{i+1:>3} {line}" for i, line in enumerate(pretty_ptx(lib.decode('utf-8')).split("\n"))]))
if DEBUG >= 6: cuda_disassemble(lib, dev.arch)
if DEBUG >= 7: cuda_disassemble(lib, dev.arch)
check(cuda.cuCtxSetCurrent(self.dev.context))
self.module = cuda.CUmodule()