From db68fa7bb14c84398fc48fc1b17fcf0cebe42704 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 8 Nov 2024 13:30:34 -0500 Subject: [PATCH] remove xla_cpu_multi_thread_eigen=false Signed-off-by: Jinzhe Zeng --- deepmd/jax/env.py | 1 - 1 file changed, 1 deletion(-) diff --git a/deepmd/jax/env.py b/deepmd/jax/env.py index 393f45319a..bdf0b8f88e 100644 --- a/deepmd/jax/env.py +++ b/deepmd/jax/env.py @@ -9,7 +9,6 @@ set_default_nthreads() inter_nthreads, intra_nthreads = get_default_nthreads() os.environ["XLA_FLAGS"] = os.environ.get("XLA_FLAGS", "") + ( - " --xla_cpu_multi_thread_eigen=false" f" intra_op_parallelism_threads={inter_nthreads}" f" inter_op_parallelism_threads={inter_nthreads}" )