diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-01-17 12:28:27 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-17 12:51:25 -0800 |
commit | 21c0fa9e2dd966c242a4e89f1cac9e3e0f146ea8 (patch) | |
tree | dafb27008ba040280825c18d10e0bef02d68187c /tensorflow/stream_executor/stream_executor_internal.h | |
parent | c276876a4ab1ebedf2b847d9ac12d019d6a16cff (diff) |
Plumb port::Status through the internal synchronous memcopy routines.
Now, at least for the public APIs that return port::Status, they can grab the
port::Status that the implementation would like to return and use its
additional information in reporting to the user.
Change: 144741667
Diffstat (limited to 'tensorflow/stream_executor/stream_executor_internal.h')
-rw-r--r-- | tensorflow/stream_executor/stream_executor_internal.h | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/tensorflow/stream_executor/stream_executor_internal.h b/tensorflow/stream_executor/stream_executor_internal.h index 57db7775a6..d6d55fd623 100644 --- a/tensorflow/stream_executor/stream_executor_internal.h +++ b/tensorflow/stream_executor/stream_executor_internal.h @@ -199,14 +199,14 @@ class StreamExecutorInterface { virtual bool SynchronousMemZero(DeviceMemoryBase *location, uint64 size) = 0; virtual bool SynchronousMemSet(DeviceMemoryBase *location, int value, uint64 size) = 0; - virtual bool SynchronousMemcpy(DeviceMemoryBase *gpu_dst, - const void *host_src, uint64 size) = 0; - virtual bool SynchronousMemcpy(void *host_dst, - const DeviceMemoryBase &gpu_src, - uint64 size) = 0; - virtual bool SynchronousMemcpyDeviceToDevice(DeviceMemoryBase *gpu_dst, - const DeviceMemoryBase &gpu_src, - uint64 size) = 0; + virtual port::Status SynchronousMemcpy(DeviceMemoryBase *gpu_dst, + const void *host_src, uint64 size) = 0; + virtual port::Status SynchronousMemcpy(void *host_dst, + const DeviceMemoryBase &gpu_src, + uint64 size) = 0; + virtual port::Status SynchronousMemcpyDeviceToDevice( + DeviceMemoryBase *gpu_dst, const DeviceMemoryBase &gpu_src, + uint64 size) = 0; virtual bool MemZero(Stream *stream, DeviceMemoryBase *location, uint64 size) = 0; virtual bool Memset(Stream *stream, DeviceMemoryBase *location, |