From be6711c02eaa390975ca7a6eb3ce92018b62272f Mon Sep 17 00:00:00 2001
From: herbakamil <herba.kamil@gmail.com>
Date: Tue, 14 Aug 2018 11:17:39 +0200
Subject: [PATCH] fixing library path in host benchmark (#164)

* fixing library path in host benchmark

* applying reviewer's comment - ensuring proper line length
---
 tools/sh_commands.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tools/sh_commands.py b/tools/sh_commands.py
index f8e09e80..d9b50342 100644
--- a/tools/sh_commands.py
+++ b/tools/sh_commands.py
@@ -976,9 +976,12 @@ def benchmark_model(abi,
     if model_graph_format == ModelFormat.file:
         mace_model_path = "%s/%s.pb" % (mace_model_dir, model_tag)
     if abi == "host":
+        libmace_dynamic_lib_dir_path = \
+            os.path.dirname(libmace_dynamic_library_path)
         p = subprocess.Popen(
             [
                 "env",
+                "LD_LIBRARY_PATH=%s" % libmace_dynamic_lib_dir_path,
                 "MACE_CPP_MIN_VLOG_LEVEL=%s" % vlog_level,
                 "%s/%s" % (benchmark_binary_dir, benchmark_binary_name),
                 "--model_name=%s" % model_tag,
-- 
GitLab