diff --git a/xla/pjrt/cpu/cpu_client.cc b/xla/pjrt/cpu/cpu_client.cc index 65ed8589f4f2b..def5ea3aaab7b 100644 --- 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()), 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(),