From 81d9053013c4cb6c7115a97b887203ceac931cb8 Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Mon, 22 Dec 2025 23:34:06 +0900 Subject: [PATCH] roc: cast to nullptr instead of changing header (#13801) --- extra/sqtt/roc.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/extra/sqtt/roc.py b/extra/sqtt/roc.py index f33afee46d..4fb9b093ad 100644 --- a/extra/sqtt/roc.py +++ b/extra/sqtt/roc.py @@ -6,15 +6,14 @@ from tinygrad.runtime.ops_amd import ProfileSQTTEvent, ProfilePMCEvent from tinygrad.runtime.autogen import llvm, rocprof from tinygrad.runtime.support.elf import elf_loader -# to pass NULL to callbacks -llvm.LLVMCreateDisasmCPUFeatures.argtypes = tuple(llvm.LLVMCreateDisasmCPUFeatures.argtypes[:5]) + (ctypes.c_void_p, ctypes.c_void_p) def llvm_disasm(arch:str, lib:bytes) -> dict[int, tuple[str, int]]: llvm.LLVMInitializeAMDGPUTargetInfo() llvm.LLVMInitializeAMDGPUTargetMC() llvm.LLVMInitializeAMDGPUAsmParser() llvm.LLVMInitializeAMDGPUDisassembler() - ctx = llvm.LLVMCreateDisasmCPUFeatures("amdgcn-amd-amdhsa".encode(), arch.encode(), "".encode(), None, 0, None, None) - + # pass NULL to callbacks + cbs = [ctypes.cast(0, llvm.LLVMCreateDisasmCPUFeatures.argtypes[i]) for i in {5,6}] + ctx = llvm.LLVMCreateDisasmCPUFeatures("amdgcn-amd-amdhsa".encode(), arch.encode(), "".encode(), None, 0, *cbs) image, sections, relocs = elf_loader(lib) text = next((sh.header for sh in sections if sh.name == ".text"), None) off, sz = unwrap(text).sh_addr, unwrap(text).sh_size