Skip to content

Commit

Permalink
fix issue BVLC#6387.
Browse files Browse the repository at this point in the history
  • Loading branch information
csukuangfj committed May 8, 2018
1 parent 8645207 commit 0536720
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions include/caffe/syncedmem.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down

0 comments on commit 0536720

Please sign in to comment.