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

Fix UnicodeDecodeError when debugging on Intel APU (#2421)

Browse Source 
* test DEBUG=5

* print prg if NVIDIA, fixes error on Intel APU
This commit is contained in:
andresgit
2023-11-25 22:30:50 +02:00
committed by GitHub
parent 7170a9a057
commit 259a869fc1
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/ops_gpu.py
View File

@@ -81,8 +81,8 @@ class CLProgram:
elif CL.cl_ctxs[0].devices[0].name.startswith('gfx'):
asm = early_exec(([ROCM_LLVM_PATH / "llvm-objdump", '-d', '-'], prg))
print('\n'.join([x for x in asm.decode('utf-8').split("\n") if 's_code_end' not in x]))
else:
# print the PTX for NVIDIA. TODO: probably broken for everything else
elif "NVIDIA" in CL.cl_ctxs[0].devices[0].name:
# print the PTX for NVIDIA.
print(prg.decode('utf-8'))
if argdtypes is not None: self.set_argdtypes(argdtypes)