Updated the multiprocess inference file

IntelLabs · Nov 1, 2023 · 4969e2b · 4969e2b
1 parent 4553842
commit 4969e2b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run_multiprocess_infer.py b/run_multiprocess_infer.py
@@ -128,9 +128,9 @@ def main(argv):
   root_condaenv=FLAGS.root_condaenv
   input_dir = FLAGS.input_dir
 
-  os.environ["LD_PRELOAD"] = "{}/lib/libiomp5.so:{}/lib/libjemalloc.so".format(root_condaenv, root_condaenv)
+  os.environ["LD_PRELOAD"] = "{}/lib/libiomp5.so:{}/lib/libjemalloc.so:$LD_PRELOAD".format(root_condaenv, root_condaenv)
   os.environ["TF_ENABLE_ONEDNN_OPTS"] = "1"
-  # os.environ["MALLOC_CONF"] = "oversize_threshold:1,background_thread:true,metadata_thp:auto,dirty_decay_ms:-1,muzzy_decay_ms:-1"
+  os.environ["MALLOC_CONF"] = "oversize_threshold:1,background_thread:true,metadata_thp:auto,dirty_decay_ms:-1,muzzy_decay_ms:-1"
   os.environ["USE_OPENMP"] = "1"
   os.environ["USE_AVX512"] = "1"
   os.environ["IPEX_ONEDNN_LAYOUT"] = "1"