feat: Add tracing feature to mnn-sync

aftershootco · Sep 24, 2024 · ae042ac · ae042ac
1 parent 7f70a7e
commit ae042ac
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -5,7 +5,7 @@ license = "Apache-2.0"
 
 [package]
 name = "mnn"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 license = { workspace = true }
 

diff --git a/mnn-sync/Cargo.toml b/mnn-sync/Cargo.toml
@@ -12,3 +12,7 @@ flume = { version = "0.11.0", default-features = false, features = [
 ] }
 mnn = { version = "0.1.0", path = ".." }
 oneshot = "0.1.8"
+tracing = { version = "0.1", optional = true }
+
+[features]
+tracing = ["dep:tracing", "mnn/tracing"]
diff --git a/mnn-sync/src/lib.rs b/mnn-sync/src/lib.rs
@@ -77,7 +77,8 @@ impl SessionHandle {
         let (tx, rx) = flume::unbounded();
         let handle = builder
             .spawn(move || -> Result<()> {
-                let session = interpreter.create_session(config)?;
+                let mut session = interpreter.create_session(config)?;
+                interpreter.update_cache_file(&mut session);
                 let mut session_runner = SessionRunner {
                     interpreter,
                     session,
@@ -103,7 +104,10 @@ impl SessionHandle {
                         if let Some(backtrace) = e.downcast_ref::<std::backtrace::Backtrace>() {
                             err = err.attach_printable(format!("{:?}", backtrace));
                         };
-                        Err(MNNError::from(err))
+                        let ret = Err(MNNError::from(err));
+                        #[cfg(feature = "tracing")]
+                        tracing::error!("Panic in session thread: {:?}", ret);
+                        ret
                     });
                     tx.send(result)
                         .change_context(ErrorKind::SyncError)