return 0;
}
google::InitGoogleLogging(argv[0]);
+ Caffe::DeviceQuery();
if (argc == 4 && strcmp(argv[3], "GPU") == 0) {
LOG(ERROR) << "Using GPU";
// cpu_data calls so that the prefetch thread does not accidentally make
// simultaneous cudaMalloc calls when the main thread is running. In some
// GPUs this seems to cause failures if we do not so.
- layer->prefetch_data_->mutable_cpu_data();
- layer->prefetch_label_->mutable_cpu_data();
+ prefetch_data_->mutable_cpu_data();
+ prefetch_label_->mutable_cpu_data();
+ data_mean_.cpu_data();
// LOG(INFO) << "Initializing prefetch";
CHECK(!pthread_create(&thread_, NULL, DataLayerPrefetch<Dtype>,
reinterpret_cast<void*>(this))) << "Pthread execution failed.";