Fix link failure in XLA unit tests

XLA unit tests were unable to link due to undefined reference to stream_executor::Stream::BlockHostUntilDone()
afzpatel · Aug 2, 2023 · de4a9a8 · de4a9a8
1 parent f0cb2a9
commit de4a9a8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/compiler/xla/service/cpu/BUILD b/tensorflow/compiler/xla/service/cpu/BUILD
@@ -397,6 +397,7 @@ cc_library(
         "//tensorflow/compiler/xla/service:executable",
         "//tensorflow/compiler/xla/service:llvm_compiler",
         "//tensorflow/compiler/xla/stream_executor",
+        "//tensorflow/compiler/xla/stream_executor:stream_executor_pimpl",  # fixdeps: keep
         "//tensorflow/compiler/xla/stream_executor/host:host_platform_id",
         "@llvm-project//llvm:Target",
     ],