Skip to content

Commit 4969e2b

Browse files
Updated the multiprocess inference file
1 parent 4553842 commit 4969e2b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

run_multiprocess_infer.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -128,9 +128,9 @@ def main(argv):
128128
root_condaenv=FLAGS.root_condaenv
129129
input_dir = FLAGS.input_dir
130130

131-
os.environ["LD_PRELOAD"] = "{}/lib/libiomp5.so:{}/lib/libjemalloc.so".format(root_condaenv, root_condaenv)
131+
os.environ["LD_PRELOAD"] = "{}/lib/libiomp5.so:{}/lib/libjemalloc.so:$LD_PRELOAD".format(root_condaenv, root_condaenv)
132132
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "1"
133-
# os.environ["MALLOC_CONF"] = "oversize_threshold:1,background_thread:true,metadata_thp:auto,dirty_decay_ms:-1,muzzy_decay_ms:-1"
133+
os.environ["MALLOC_CONF"] = "oversize_threshold:1,background_thread:true,metadata_thp:auto,dirty_decay_ms:-1,muzzy_decay_ms:-1"
134134
os.environ["USE_OPENMP"] = "1"
135135
os.environ["USE_AVX512"] = "1"
136136
os.environ["IPEX_ONEDNN_LAYOUT"] = "1"

0 commit comments

Comments
 (0)