[xla:cpu] Use default ThreadOptions for eigen intra-op thread pool

PiperOrigin-RevId: 666109330
openxla · Aug 22, 2024 · 0fbf2e1 · 0fbf2e1
1 parent e6667b5
commit 0fbf2e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xla/pjrt/cpu/cpu_client.cc b/xla/pjrt/cpu/cpu_client.cc
@@ -421,7 +421,7 @@ TfrtCpuClient::TfrtCpuClient(
       owned_devices_(std::move(devices)),
       computation_placer_(std::make_unique<ComputationPlacer>()),
       eigen_intraop_pool_(new tsl::thread::ThreadPool(
-          tsl::Env::Default(), "XLAEigen",
+          tsl::Env::Default(), GetThreadOptions(), "XLAEigen",
           std::min(num_threads, kMaxIntraOpThreads))),
       eigen_intraop_device_(
           new Eigen::ThreadPoolDevice(eigen_intraop_pool_->AsEigenThreadPool(),