diff --git a/tinygrad/runtime/ops_qcom.py b/tinygrad/runtime/ops_qcom.py index e5e2183a..2f16144e 100644 --- a/tinygrad/runtime/ops_qcom.py +++ b/tinygrad/runtime/ops_qcom.py @@ -6,7 +6,7 @@ from tinygrad.device import BufferOptions, HCQBuffer, HWComputeQueue, HCQProgram from tinygrad.runtime.autogen import kgsl, adreno, libc from tinygrad.runtime.ops_gpu import CLCompiler, CLDevice from tinygrad.renderer.cstyle import QCOMRenderer -from tinygrad.helpers import getenv, from_mv, mv_address, to_mv, round_up, data64_le, prod +from tinygrad.helpers import getenv, from_mv, mv_address, to_mv, round_up, data64_le, prod, DEBUG, fromimport if getenv("IOCTL"): import extra.qcom_gpu_driver.opencl_ioctl # noqa: F401 # pylint: disable=unused-import def _qreg_exec(reg, __val=0, **kwargs): @@ -219,6 +219,7 @@ class QCOMArgsState(HCQArgsState): class QCOMProgram(HCQProgram): def __init__(self, device: QCOMDevice, name: str, lib: bytes): self.device, self.name, self.lib = device, name, lib + if DEBUG >= 5: fromimport('extra.disassemblers.adreno', 'disasm')(lib) self._parse_lib() self.lib_gpu = self.device.allocator.alloc(self.image_size, options=BufferOptions(cpu_access=True, nolru=True))