Increase device count to support 2x2x2 topology

PiperOrigin-RevId: 679780226
openxla · Sep 28, 2024 · e0dea55 · e0dea55
1 parent 3c5c920
commit e0dea55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xla/python/pjrt_ifrt/tfrt_cpu_client_test_lib.cc b/xla/python/pjrt_ifrt/tfrt_cpu_client_test_lib.cc
@@ -27,7 +27,7 @@ namespace {
 const bool kUnused = (test_util::RegisterClientFactory(
                           []() -> absl::StatusOr<std::shared_ptr<Client>> {
                             CpuClientOptions options;
-                            options.cpu_device_count = 4;
+                            options.cpu_device_count = 8;
                             TF_ASSIGN_OR_RETURN(auto pjrt_client,
                                                 xla::GetTfrtCpuClient(options));
                             return std::shared_ptr<Client>(