Merge pull request #6 from tenstorrent/ddilbaz/closedevice

Close device to stop seg fault
tenstorrent · Oct 29, 2024 · 99a3ebc · 99a3ebc
2 parents 1a4da93 + 158768e
commit 99a3ebc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tt_torch/csrc/bindings.cpp b/tt_torch/csrc/bindings.cpp
@@ -104,8 +104,9 @@ std::vector<at::Tensor> run(const std::vector<at::Tensor>& inputs, tt::runtime::
         rt_outputs.emplace_back(create_tensor(outputs.back()));
     }
 
-    tt::runtime::Event _ = tt::runtime::submit(device, binary, program_idx, rt_inputs, rt_outputs);
-
+    tt::runtime::Event event = tt::runtime::submit(device, binary, program_idx, rt_inputs, rt_outputs);
+    (void)event;
+    tt::runtime::closeDevice(device);
     return outputs;
 }