fix auto tune issue

mehrdadh · Aug 11, 2021 · 25d2969 · 25d2969
1 parent 4396519
commit 25d2969
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/micro/session.py b/python/tvm/micro/session.py
@@ -272,6 +272,7 @@ def compile_and_create_micro_session(
     global RPC_SESSION
 
     temp_dir = utils.tempdir()
+    temp_dir.set_keep_for_debug(True)
     model_library_format_path = temp_dir / "model.tar.gz"
     print("PRE GEN", model_library_format_path)
     with open(model_library_format_path, "wb") as mlf_f: