Skip to content

Commit 0e324a7

Browse files
Automated Code Change
PiperOrigin-RevId: 722187263
1 parent 8bcd7e7 commit 0e324a7

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

tensorflow/core/common_runtime/next_pluggable_device/c/BUILD

+3
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ tf_cc_test(
4545
"//tensorflow/c:c_api",
4646
"//tensorflow/c:tf_status_headers",
4747
"//tensorflow/core/platform:status",
48+
"@com_google_absl//absl/log",
49+
"@com_google_absl//absl/status",
4850
"@com_google_googletest//:gtest_main",
4951
"@tf_runtime//:hostcontext_alwayslink",
5052
],
@@ -170,6 +172,7 @@ tf_cc_test(
170172
"@com_google_absl//absl/status",
171173
"@com_google_absl//absl/strings",
172174
"@com_google_absl//absl/synchronization",
175+
"@com_google_absl//absl/types:span",
173176
"@com_google_googletest//:gtest_main",
174177
"@local_tsl//tsl/platform:status",
175178
"@local_xla//xla/tsl/lib/core:status_test_util",

tensorflow/core/common_runtime/next_pluggable_device/c/plugin_c_api_test.cc

+2
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,8 @@ limitations under the License.
2121
#include <utility>
2222

2323
#include <gtest/gtest.h>
24+
#include "absl/log/log.h"
25+
#include "absl/status/status.h"
2426
#include "tensorflow/c/tf_status.h"
2527
#include "tensorflow/core/common_runtime/next_pluggable_device/c/example_plugin.h"
2628
#include "tensorflow/core/platform/status.h"

tensorflow/core/common_runtime/next_pluggable_device/c/tf_rendezvous_c_api_test.cc

+1
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ limitations under the License.
2525
#include "absl/status/status.h"
2626
#include "absl/strings/str_cat.h"
2727
#include "absl/synchronization/notification.h"
28+
#include "absl/types/span.h"
2829
#include "xla/tsl/framework/allocator.h"
2930
#include "xla/tsl/lib/core/status_test_util.h"
3031
#include "tensorflow/core/common_runtime/device_mgr.h"

0 commit comments

Comments
 (0)