| 82 | } |
| 83 | |
| 84 | void SyncedMemory::set_cpu_data(void* data) { |
| 85 | CHECK(data); |
| 86 | if (own_cpu_data_) { |
| 87 | CaffeFreeHost(cpu_ptr_, cpu_malloc_use_cuda_); |
| 88 | } |
| 89 | cpu_ptr_ = data; |
| 90 | head_ = HEAD_AT_CPU; |
| 91 | own_cpu_data_ = false; |
| 92 | } |
| 93 | |
| 94 | const void* SyncedMemory::gpu_data() { |
| 95 | #ifndef CPU_ONLY |
nothing calls this directly
no test coverage detected