cudaSetDevice(device);
cout << "Setting to use device " << device << endl;
} else if (CUDA_TEST_DEVICE >= 0) {
-
- // Use the device assigned in build configuration; but with a lower priority
- device = CUDA_TEST_DEVICE;
+ // Use the device assigned in build configuration; but with a lower priority
+ device = CUDA_TEST_DEVICE;
}
cudaGetDevice(&device);
cout << "Current device id: " << device << endl;
protected:
HDF5OutputLayerTest()
: output_file_name_(tmpnam(NULL)),
- input_file_name_(CMAKE_SOURCE_DIR "caffe/test/test_data/sample_data.h5"),
+ input_file_name_(
+ CMAKE_SOURCE_DIR "caffe/test/test_data/sample_data.h5"),
blob_data_(new Blob<Dtype>()),
blob_label_(new Blob<Dtype>()),
num_(5),
blob_top_vec_.push_back(blob_top_label_);
// Check out generate_sample_data.py in the same directory.
- filename = new string(CMAKE_SOURCE_DIR "caffe/test/test_data/sample_data_list.txt" CMAKE_EXT);
- LOG(INFO) << "Using sample HDF5 data file " << filename;
+ filename = new string(
+ CMAKE_SOURCE_DIR "caffe/test/test_data/sample_data_list.txt" CMAKE_EXT);
+ LOG(INFO)<< "Using sample HDF5 data file " << filename;
}
virtual ~HDF5DataLayerTest() {