bring back TINYGRAD_DEBUG. not sure why it was removed

980d5d2c5b681e563f5577da7b3c219904e4018e · 2024-07-31 22:56:53 +0100 · Alex Cheema

Files touched

M exo/helpers.py

Diff

commit 980d5d2c5b681e563f5577da7b3c219904e4018e
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Wed Jul 31 22:56:53 2024 +0100

    bring back TINYGRAD_DEBUG. not sure why it was removed
---
 exo/helpers.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/exo/helpers.py b/exo/helpers.py
index a46b840e..71281a35 100644
--- a/exo/helpers.py
+++ b/exo/helpers.py
@@ -36,6 +36,8 @@ def get_inference_engine(inference_engine_name):
     return MLXDynamicShardInferenceEngine()
   elif inference_engine_name == "tinygrad":
     from exo.inference.tinygrad.inference import TinygradDynamicShardInferenceEngine
+    import tinygrad.helpers
+    tinygrad.helpers.DEBUG.value = int(os.getenv("TINYGRAD_DEBUG", default="0"))
 
     return TinygradDynamicShardInferenceEngine()
   else:

← 76766253 fix regression introduced by image_str for tinygrad · back to Exo · sticky node ids #16 0bfb8e3b →