diff options
Diffstat (limited to 'src/caffe/test')
-rw-r--r-- | src/caffe/test/test_caffe_main.cpp | 5 | ||||
-rw-r--r-- | src/caffe/test/test_hdf5_output_layer.cpp | 3 | ||||
-rw-r--r-- | src/caffe/test/test_hdf5data_layer.cpp | 5 |
3 files changed, 7 insertions, 6 deletions
diff --git a/src/caffe/test/test_caffe_main.cpp b/src/caffe/test/test_caffe_main.cpp index ad6e189a..bff0c4ed 100644 --- a/src/caffe/test/test_caffe_main.cpp +++ b/src/caffe/test/test_caffe_main.cpp @@ -27,9 +27,8 @@ int main(int argc, char** argv) { 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; diff --git a/src/caffe/test/test_hdf5_output_layer.cpp b/src/caffe/test/test_hdf5_output_layer.cpp index f3ddfcd9..b2a511d1 100644 --- a/src/caffe/test/test_hdf5_output_layer.cpp +++ b/src/caffe/test/test_hdf5_output_layer.cpp @@ -23,7 +23,8 @@ class HDF5OutputLayerTest : public MultiDeviceTest<TypeParam> { 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), diff --git a/src/caffe/test/test_hdf5data_layer.cpp b/src/caffe/test/test_hdf5data_layer.cpp index 1de82330..acca75b1 100644 --- a/src/caffe/test/test_hdf5data_layer.cpp +++ b/src/caffe/test/test_hdf5data_layer.cpp @@ -31,8 +31,9 @@ class HDF5DataLayerTest : public MultiDeviceTest<TypeParam> { 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() { |