#287: Hotfix for seg fault when executing dir of flatbuffers

tenstorrent · Aug 2, 2024 · 16b592e · 16b592e
1 parent dcb4b51
commit 16b592e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/runtime/tools/python/ttrt/common/api.py b/runtime/tools/python/ttrt/common/api.py
@@ -38,7 +38,7 @@ def read(args):
     arg_binary = args.binary
     arg_clean_artifacts = args.clean_artifacts
     arg_save_artifacts = args.save_artifacts
-    arg_section = arg.section
+    arg_section = args.section
 
     # preprocessing
     if os.path.isdir(arg_binary):
@@ -182,11 +182,13 @@ def run(args):
 
         system_desc, device_ids = ttrt.runtime.get_current_system_desc()
         device = ttrt.runtime.open_device(device_ids)
-        atexit.register(lambda: ttrt.runtime.close_device(device))
+        #atexit.register(lambda: ttrt.runtime.close_device(device))
         for loop in range(arg_loops):
             ttrt.runtime.submit(device, fbb, 0, total_inputs[loop], total_outputs[loop])
             print(f"finished loop={loop}")
         print("outputs:\n", torch_outputs)
+        ttrt.runtime.close_device(device)
+
 
     # save artifacts
     if arg_save_artifacts: