C++ (Cpp) to_gpu Examples

Example #1

File: syncedmem.cpp Project: dongxf369/Dragon

void* SyncedMemory::mutable_gpu_data(){
#ifndef CPU_ONLY
	to_gpu();
	head_ = HEAD_AT_GPU;
	return gpu_ptr;
#endif
}

Example #2

File: syncedmem.cpp Project: sri-kankanahalli/caffe

const void* SyncedMemory::gpu_data() {
#ifndef CPU_ONLY
  to_gpu();
  return (const void*)gpu_ptr_;
#else
  NO_GPU;
#endif
}

Example #3

File: syncedmem.cpp Project: flair2005/Caffe-Solution

void* SyncedMemory::mutable_gpu_data() {
#ifndef CPU_ONLY
  to_gpu();
  head_ = HEAD_AT_GPU;
  return gpu_ptr_;
#else
  NO_GPU;
  return 0;//Modified by Jianfeng
#endif
}

Example #4

File: syncedmem.cpp Project: xiaomi646/caffe_2d_mlabel

void* SyncedMemory::mutable_gpu_data() {
  // LOG(INFO)<<"calling mutable_gpu_data";
  // switch (head_) {
  // case UNINITIALIZED:
	// LOG(INFO)<<"MEM not initilaized";
	// break;
  // case HEAD_AT_CPU:
     // LOG(INFO)<<"MEM is in CPU";
	// break;
  // case HEAD_AT_GPU:
    // LOG(INFO)<<"MEM is in GPU";
	// break;
	// }
  to_gpu();
  head_ = HEAD_AT_GPU;
  return gpu_ptr_;
}

Example #5

File: syncedmem.cpp Project: AlOa/caffe

void* SyncedMemory::mutable_gpu_data() {
  to_gpu();
  head_ = HEAD_AT_GPU;
  return gpu_ptr_;
}

Example #6

File: syncedmem.cpp Project: AlOa/caffe

const void* SyncedMemory::gpu_data() {
  to_gpu();
  return (const void*)gpu_ptr_;
}