diff --git a/run_multiprocess_infer.py b/run_multiprocess_infer.py index 7273c58..1fdabfd 100644 --- 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"