feat: impl drop for interpreter

aftershootco · Sep 21, 2024 · ed12f3d · ed12f3d
1 parent ceb83d3
commit ed12f3d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/flake.nix b/flake.nix
@@ -171,8 +171,8 @@
                 cargo-nextest
                 cargo-hakari
                 cargo-deny
-                rust-cbindgen
                 cargo-semver-checks
+                rust-bindgen
               ]
               ++ (lib.optionals pkgs.stdenv.isDarwin [
                 darwin.apple_sdk.frameworks.OpenCL

diff --git a/src/interpreter.rs b/src/interpreter.rs
@@ -129,6 +129,12 @@ pub struct Interpreter {
 
 unsafe impl Send for Interpreter {}
 
+impl Drop for Interpreter {
+    fn drop(&mut self) {
+        unsafe { mnn_sys::Interpreter_destroy(self.inner) }
+    }
+}
+
 impl Interpreter {
     pub fn from_file(path: impl AsRef<Path>) -> Result<Self> {
         let path = path.as_ref();
@@ -636,3 +642,4 @@ fn try_to_drop_interpreter_before_session() {
     drop(interpreter);
     drop(session);
 }
+