From 5316ca0a06efe71eed59dc4eef1f17606c8f2b56 Mon Sep 17 00:00:00 2001 From: Sergio Date: Sat, 28 Jun 2014 10:22:13 -0700 Subject: [PATCH] Remove Cuda.major >= 2 check on Dropout test --- src/caffe/test/test_neuron_layer.cpp | 8 -------- 1 file changed, 8 deletions(-) diff --git a/src/caffe/test/test_neuron_layer.cpp b/src/caffe/test/test_neuron_layer.cpp index c7c0dc1..a439128 100644 --- a/src/caffe/test/test_neuron_layer.cpp +++ b/src/caffe/test/test_neuron_layer.cpp @@ -226,7 +226,6 @@ TYPED_TEST(NeuronLayerTest, TestDropoutGPU) { TYPED_TEST(NeuronLayerTest, TestDropoutGradientGPU) { - if (CAFFE_TEST_CUDA_PROP.major >= 2) { LayerParameter layer_param; Caffe::set_mode(Caffe::GPU); Caffe::set_phase(Caffe::TRAIN); @@ -236,13 +235,9 @@ TYPED_TEST(NeuronLayerTest, TestDropoutGradientGPU) { // exhaustive gradient check. checker.CheckGradient(&layer, &(this->blob_bottom_vec_), &(this->blob_top_vec_)); - } else { - LOG(ERROR) << "Skipping test to spare my laptop."; - } } TYPED_TEST(NeuronLayerTest, TestDropoutGradientGPUTest) { - if (CAFFE_TEST_CUDA_PROP.major >= 2) { LayerParameter layer_param; Caffe::set_mode(Caffe::GPU); Caffe::set_phase(Caffe::TEST); @@ -252,9 +247,6 @@ TYPED_TEST(NeuronLayerTest, TestDropoutGradientGPUTest) { // exhaustive gradient check. checker.CheckGradient(&layer, &(this->blob_bottom_vec_), &(this->blob_top_vec_)); - } else { - LOG(ERROR) << "Skipping test to spare my laptop."; - } } -- 2.7.4