From b8a74c15696f2ef3a3fc0b4514da7cbed6873805 Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Mon, 15 Sep 2025 13:29:44 +0300 Subject: [PATCH] cpu: add disassembler err message (#12184) * cpu: add disassembler err message * print msg --- tinygrad/helpers.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tinygrad/helpers.py b/tinygrad/helpers.py index 82fe093f6e..95628c300a 100644 --- a/tinygrad/helpers.py +++ b/tinygrad/helpers.py @@ -325,7 +325,10 @@ def cpu_objdump(lib, objdump_tool='objdump'): print(subprocess.check_output([objdump_tool, '-d', f.name]).decode('utf-8')) def capstone_flatdump(lib: bytes): - import capstone + try: import capstone + except ImportError: + print("Disassembler Error: Capstone not installed.") + return match platform.machine(): case 'x86_64' | 'AMD64': cs = capstone.Cs(capstone.CS_ARCH_X86, capstone.CS_MODE_64) case 'aarch64' | 'arm64': cs = capstone.Cs(capstone.CS_ARCH_ARM64, capstone.CS_MODE_ARM)