remove xla_cpu_multi_thread_eigen=false

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Nov 8, 2024 · db68fa7 · db68fa7
1 parent 00e3c10
commit db68fa7
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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}"
 )