| 119 | } |
| 120 | |
| 121 | void SyncMem::set_device_data(void *data) { |
| 122 | #ifdef USE_CUDA |
| 123 | CHECK_NOTNULL(data); |
| 124 | if (own_device_data) { |
| 125 | CUDA_CHECK(cudaFree(device_data())); |
| 126 | total_memory_size -= size_; |
| 127 | } |
| 128 | device_ptr = data; |
| 129 | own_device_data = false; |
| 130 | head_ = HEAD::DEVICE; |
| 131 | #else |
| 132 | NO_GPU; |
| 133 | #endif |
| 134 | } |
| 135 | } |