nv ioctl sniffer (#3892)

* nv ioctl sniffer

* unused import

* Update __init__.py

* that work

* that fix it

tinygrad/runtime/ops_cuda.py

@@ -8,6 +8,7 @@ from tinygrad.device import Compiled, LRUAllocator, MallocAllocator, Compiler, B
 from tinygrad.codegen.kernel import LinearizerOptions
 from tinygrad.renderer.cstyle import CUDARenderer
 from tinygrad.renderer.assembly import PTXRenderer
+if getenv("IOCTL"): import extra.nv_gpu_driver.nv_ioctl  # noqa: F401
 
 def pretty_ptx(s):
   # all expressions match `<valid_before><expr><valid_after>` and replace it with `<valid_before>color(<expr>)<valid_after>`

tinygrad/runtime/ops_hsa.py

@@ -8,6 +8,7 @@ from tinygrad.codegen.kernel import LinearizerOptions
 from tinygrad.runtime.driver.hsa import check, scan_agents, find_memory_pool, AQLQueue
 from tinygrad.renderer.cstyle import HIPRenderer
 from tinygrad.runtime.driver.hip_comgr import compile_hip
+if getenv("IOCTL"): import extra.hip_gpu_driver.hip_ioctl  # noqa: F401
 
 PROFILE = getenv("PROFILE", 0)