From a66e168ed76d301dadedc8299bffd94c7e398e3b Mon Sep 17 00:00:00 2001
From: Zach Kimberg <kimbergz@amazon.com>
Date: Mon, 8 Mar 2021 12:58:11 -0600
Subject: [PATCH] Configure verbose of mxnet extra libraries (#728)

Change-Id: I66d54aa496cccbb9e8c0a89eeaa458605958d9c6
---
 .../src/main/java/ai/djl/mxnet/engine/MxEngine.java         | 6 +++++-
 .../src/main/java/ai/djl/mxnet/jna/JnaUtils.java            | 5 +++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxEngine.java b/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxEngine.java
index 2106c8d2f52..527a1d1f693 100644
--- a/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxEngine.java
+++ b/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxEngine.java
@@ -40,6 +40,7 @@
 public final class MxEngine extends Engine {
 
     public static final String ENGINE_NAME = "MXNet";
+    private static final String MXNET_EXTRA_LIBRARY_VERBOSE = "MXNET_EXTRA_LIBRARY_VERBOSE";
 
     /** Constructs an MXNet Engine. */
     private MxEngine() {}
@@ -56,6 +57,9 @@ static Engine newInstance() {
 
             // load extra MXNet library
             String paths = System.getenv("MXNET_EXTRA_LIBRARY_PATH");
+            boolean extraLibVerbose =
+                    System.getenv().containsKey(MXNET_EXTRA_LIBRARY_VERBOSE)
+                            && System.getenv(MXNET_EXTRA_LIBRARY_VERBOSE).equals("true");
             if (paths != null) {
                 String[] files = paths.split(",");
                 for (String file : files) {
@@ -63,7 +67,7 @@ static Engine newInstance() {
                     if (Files.notExists(path)) {
                         throw new FileNotFoundException("Extra Library not found: " + file);
                     }
-                    JnaUtils.loadLib(path.toAbsolutePath().toString(), 1);
+                    JnaUtils.loadLib(path.toAbsolutePath().toString(), extraLibVerbose);
                 }
             }
 
diff --git a/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/jna/JnaUtils.java b/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/jna/JnaUtils.java
index 0527049a25a..43daa3e388f 100644
--- a/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/jna/JnaUtils.java
+++ b/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/jna/JnaUtils.java
@@ -1254,8 +1254,9 @@ public static List<List<Shape>> inferShape(Symbol symbol, PairList<String, Shape
         return null;
     }
 
-    public static void loadLib(String path, int verbose) {
-        checkCall(LIB.MXLoadLib(path, verbose));
+    public static void loadLib(String path, boolean verbose) {
+        int intVerbose = verbose ? 1 : 0;
+        checkCall(LIB.MXLoadLib(path, intVerbose));
     }
 
     public static Pointer optimizeFor(Symbol current, String backend, Device device) {