From 6cc0ef8afe751164c17ab4981ba64957685594ef Mon Sep 17 00:00:00 2001 From: CUI Wei Date: Sun, 29 Jul 2018 20:54:54 -0400 Subject: Fix typo: host_src -> gpu_src for inter-gpu copy Signed-off-by: CUI Wei --- tensorflow/stream_executor/stream_executor_internal.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/stream_executor') diff --git a/tensorflow/stream_executor/stream_executor_internal.h b/tensorflow/stream_executor/stream_executor_internal.h index f34b1fc083..52b3dc04c4 100644 --- a/tensorflow/stream_executor/stream_executor_internal.h +++ b/tensorflow/stream_executor/stream_executor_internal.h @@ -236,7 +236,7 @@ class StreamExecutorInterface { virtual bool Memcpy(Stream *stream, DeviceMemoryBase *gpu_dst, const void *host_src, uint64 size) = 0; virtual bool MemcpyDeviceToDevice(Stream *stream, DeviceMemoryBase *gpu_dst, - const DeviceMemoryBase &host_src, + const DeviceMemoryBase &gpu_src, uint64 size) = 0; virtual bool HostCallback(Stream *stream, std::function callback) = 0; virtual port::Status AllocateEvent(Event *event) = 0; -- cgit v1.2.3