Skip to content

Commit 0fbf2e1

Browse files
ezhulenevcopybara-github
authored andcommitted
[xla:cpu] Use default ThreadOptions for eigen intra-op thread pool
PiperOrigin-RevId: 666109330
1 parent e6667b5 commit 0fbf2e1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

xla/pjrt/cpu/cpu_client.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -421,7 +421,7 @@ TfrtCpuClient::TfrtCpuClient(
421421
owned_devices_(std::move(devices)),
422422
computation_placer_(std::make_unique<ComputationPlacer>()),
423423
eigen_intraop_pool_(new tsl::thread::ThreadPool(
424-
tsl::Env::Default(), "XLAEigen",
424+
tsl::Env::Default(), GetThreadOptions(), "XLAEigen",
425425
std::min(num_threads, kMaxIntraOpThreads))),
426426
eigen_intraop_device_(
427427
new Eigen::ThreadPoolDevice(eigen_intraop_pool_->AsEigenThreadPool(),

0 commit comments

Comments
 (0)