diff options
author | CUI Wei <ghostplant@qq.com> | 2018-07-29 20:54:54 -0400 |
---|---|---|
committer | CUI Wei <ghostplant@qq.com> | 2018-07-29 20:54:54 -0400 |
commit | 6cc0ef8afe751164c17ab4981ba64957685594ef (patch) | |
tree | f5cdd03f04fc2b0a803e436c3dc6d8a514447d4e /tensorflow/stream_executor | |
parent | e7475504094b3018973740df470d2c9ee73a4fd5 (diff) |
Fix typo: host_src -> gpu_src for inter-gpu copy
Signed-off-by: CUI Wei <ghostplant@qq.com>
Diffstat (limited to 'tensorflow/stream_executor')
-rw-r--r-- | tensorflow/stream_executor/stream_executor_internal.h | 2 |
1 files changed, 1 insertions, 1 deletions
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<void()> callback) = 0; virtual port::Status AllocateEvent(Event *event) = 0; |