| 108 | } |
| 109 | |
| 110 | void SyncMem::set_host_data(void *data) { |
| 111 | CHECK_NOTNULL(data); |
| 112 | if (own_host_data) { |
| 113 | free_host(host_ptr); |
| 114 | total_memory_size -= size_; |
| 115 | } |
| 116 | host_ptr = data; |
| 117 | own_host_data = false; |
| 118 | head_ = HEAD::HOST; |
| 119 | } |
| 120 | |
| 121 | void SyncMem::set_device_data(void *data) { |
| 122 | #ifdef USE_CUDA |