projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
c6d86f7
)
fix issue #6387.
author
Kuang Fangjun
<csukuangfj@gmail.com>
Tue, 8 May 2018 02:36:44 +0000
(10:36 +0800)
committer
Wook Song
<wook16.song@samsung.com>
Thu, 23 Jan 2020 13:50:42 +0000
(22:50 +0900)
include/caffe/syncedmem.hpp
patch
|
blob
|
history
diff --git
a/include/caffe/syncedmem.hpp
b/include/caffe/syncedmem.hpp
index
317ce29
..
8d650a3
100644
(file)
--- a/
include/caffe/syncedmem.hpp
+++ b/
include/caffe/syncedmem.hpp
@@
-66,8
+66,8
@@
class SyncedMemory {
void* mutable_cpu_data();
void* mutable_gpu_data();
enum SyncedHead { UNINITIALIZED, HEAD_AT_CPU, HEAD_AT_GPU, SYNCED };
- SyncedHead head() { return head_; }
- size_t size() { return size_; }
+ SyncedHead head()
const
{ return head_; }
+ size_t size()
const
{ return size_; }
#ifndef CPU_ONLY
void async_gpu_push(const cudaStream_t& stream);