Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Automated Code Change
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 718630616
Google-ML-Automation committed Jan 23, 2025
1 parent 5b382c1 commit 9dc2055
Showing 5 changed files with 6 additions and 0 deletions.
2 changes: 2 additions & 0 deletions xla/pjrt/plugin/xla_cpu/BUILD
Original file line number Diff line number Diff line change
@@ -30,6 +30,7 @@ xla_cc_test(
"//xla/pjrt:pjrt_client",
"//xla/pjrt/cpu:cpu_client",
"//xla/tests:xla_internal_test_main",
"@com_google_googletest//:gtest",
"@tsl//tsl/platform:statusor",
"@tsl//tsl/platform:test",
],
@@ -67,6 +68,7 @@ cc_library(
":cpu_device_description",
":cpu_topology",
"//xla:shape_util",
"//xla:xla_data_proto_cc",
"//xla/pjrt:pjrt_client",
"//xla/pjrt:pjrt_compiler",
"//xla/pjrt:pjrt_device_description",
1 change: 1 addition & 0 deletions xla/pjrt/plugin/xla_cpu/cpu_topology_description.cc
Original file line number Diff line number Diff line change
@@ -34,6 +34,7 @@ limitations under the License.
#include "xla/shape.h"
#include "xla/shape_util.h"
#include "xla/tsl/lib/strings/proto_serialization.h"
#include "xla/xla_data.pb.h"

namespace xla {

1 change: 1 addition & 0 deletions xla/pjrt/plugin/xla_cpu/cpu_topology_description.h
Original file line number Diff line number Diff line change
@@ -31,6 +31,7 @@ limitations under the License.
#include "xla/pjrt/pjrt_compiler.h"
#include "xla/pjrt/pjrt_device_description.h"
#include "xla/pjrt/plugin/xla_cpu/cpu_topology.h"
#include "xla/xla_data.pb.h"

namespace xla {

1 change: 1 addition & 0 deletions xla/pjrt/plugin/xla_cpu/cpu_topology_test.cc
Original file line number Diff line number Diff line change
@@ -17,6 +17,7 @@ limitations under the License.

#include <memory>

#include <gtest/gtest.h>
#include "xla/pjrt/plugin/xla_cpu/cpu_topology.pb.h"
#include "tsl/platform/protobuf.h"
#include "tsl/platform/test.h"
1 change: 1 addition & 0 deletions xla/pjrt/plugin/xla_cpu/xla_cpu_pjrt_client_test.cc
Original file line number Diff line number Diff line change
@@ -15,6 +15,7 @@ limitations under the License.

#include "xla/pjrt/plugin/xla_cpu/xla_cpu_pjrt_client.h"

#include <gtest/gtest.h>
#include "xla/pjrt/cpu/cpu_client.h"
#include "xla/pjrt/pjrt_client.h"
#include "tsl/platform/statusor.h"

0 comments on commit 9dc2055

Please sign in to comment.