$ORT_ALLOW_VERBOSE_LOGGING_ON_RELEASEを導入

onnxruntime/build.rs

@@ -0,0 +1,14 @@
+use std::env;
+
+fn main() {
+    if boolean_env_var("ORT_ALLOW_DEBUG_LOGGING") {
+        println!("cargo:rustc-cfg=allow_debug_logging");
+    }
+    println!("cargo:rerun-if-env-changed=ORT_ALLOW_DEBUG_LOGGING");
+}
+
+fn boolean_env_var(name: &str) -> bool {
+    // Same as `ORT_USE_CUDA`.
+    let var = env::var(name).unwrap_or_default();
+    matches!(&*var.to_lowercase(), "1" | "yes" | "true" | "on")
+}

onnxruntime/src/environment.rs

@@ -6,7 +6,7 @@ use std::{
 };
 
 use lazy_static::lazy_static;
-use tracing::{debug, error, warn};
+use tracing::{error, warn};
 
 use onnxruntime_sys as sys;
 

onnxruntime/src/lib.rs

@@ -145,6 +145,26 @@ macro_rules! extern_system_fn {
     ($(#[$meta:meta])* $vis:vis unsafe fn $($tt:tt)*) => ($(#[$meta])* $vis unsafe extern "C" fn $($tt)*);
 }
 
+// textual scopeで`trace!`と`debug!`を宣言することにより、`tracing`のそれらの`use`を封じる
+
+macro_rules! trace {
+    ($($tt:tt)*) => {
+        if cfg!(allow_debug_logging) {
+            ::tracing::trace!($($tt)*)
+        } else {
+        }
+    };
+}
+
+macro_rules! debug {
+    ($($tt:tt)*) => {
+        if cfg!(allow_debug_logging) {
+            ::tracing::debug!($($tt)*)
+        } else {
+        }
+    };
+}
+
 pub mod download;
 pub mod environment;
 pub mod error;
@@ -205,7 +225,7 @@ mod onnxruntime {
     //! to Rust's tracing logging instead.
 
     use std::ffi::CStr;
-    use tracing::{debug, error, info, span, trace, warn, Level};
+    use tracing::{error, info, span, warn, Level};
 
     use onnxruntime_sys as sys;
 
@@ -271,16 +291,18 @@ mod onnxruntime {
             // Parse the code location
             let code_location: CodeLocation = code_location.into();
 
-            let span = span!(
-                Level::TRACE,
-                "onnxruntime",
-                category = category.to_str().unwrap_or("<unknown>"),
-                file = code_location.file,
-                line_number = code_location.line_number,
-                function = code_location.function,
-                logid = logid.to_str().unwrap_or("<unknown>"),
-            );
-            let _enter = span.enter();
+            let _span = cfg!(allow_verbose_logging).then(|| {
+                span!(
+                    Level::TRACE,
+                    "onnxruntime",
+                    category = category.to_str().unwrap_or("<unknown>"),
+                    file = code_location.file,
+                    line_number = code_location.line_number,
+                    function = code_location.function,
+                    logid = logid.to_str().unwrap_or("<unknown>"),
+                )
+                .entered()
+            });
 
             match log_level {
                 Level::TRACE => trace!("{:?}", message),

onnxruntime/src/memory.rs

@@ -1,5 +1,3 @@
-use tracing::debug;
-
 use onnxruntime_sys as sys;
 
 use tracing::error;

onnxruntime/src/session.rs

@@ -17,7 +17,7 @@ use std::os::windows::ffi::OsStrExt;
 use std::env;
 
 use ndarray::Array;
-use tracing::{debug, error};
+use tracing::error;
 
 use onnxruntime_sys as sys;
 

onnxruntime/src/tensor/ort_owned_tensor.rs

@@ -3,7 +3,6 @@
 use std::{fmt::Debug, ops::Deref};
 
 use ndarray::{Array, ArrayView};
-use tracing::debug;
 
 use onnxruntime_sys as sys;
 

onnxruntime/src/tensor/ort_tensor.rs

@@ -3,7 +3,7 @@
 use std::{ffi, fmt::Debug, ops::Deref};
 
 use ndarray::Array;
-use tracing::{debug, error};
+use tracing::error;
 
 use onnxruntime_sys as sys;