[XLA GPU] Remove redundant dependency
PiperOrigin-RevId: 253326367
This commit is contained in:
parent
dc43ecaa93
commit
19c72dd784
@ -1194,7 +1194,6 @@ cc_library(
|
||||
srcs = ["buffer_comparator.cc"],
|
||||
hdrs = ["buffer_comparator.h"],
|
||||
deps = [
|
||||
":gpu_executable",
|
||||
":partition_assignment",
|
||||
":stream_executor_util",
|
||||
"//tensorflow/compiler/xla:shape_util",
|
||||
@ -1217,6 +1216,7 @@ tf_cc_test(
|
||||
"//tensorflow/compiler/xla:types",
|
||||
"//tensorflow/core:test",
|
||||
"//tensorflow/core:test_main",
|
||||
"//tensorflow/core/platform/default/build_config:stream_executor_cuda", # build_cleaner: keep
|
||||
"//tensorflow/stream_executor:device_memory",
|
||||
],
|
||||
)
|
||||
|
@ -16,7 +16,6 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_GPU_BUFFER_COMPARATOR_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_GPU_BUFFER_COMPARATOR_H_
|
||||
|
||||
#include "tensorflow/compiler/xla/service/gpu/gpu_executable.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_module_config.h"
|
||||
#include "tensorflow/compiler/xla/shape.h"
|
||||
#include "tensorflow/core/platform/stream_executor_no_cuda.h"
|
||||
|
Loading…
x
Reference in New Issue
Block a user