From c4e4ca82e47c37ed1ef09cf94a43153bd5a0c374 Mon Sep 17 00:00:00 2001 From: Sergio Date: Tue, 27 May 2014 11:44:17 -0700 Subject: [PATCH] Comment Test GPUs cases --- src/caffe/test/test_threshold_layer.cpp | 92 ++++++++++++++++----------------- 1 file changed, 46 insertions(+), 46 deletions(-) diff --git a/src/caffe/test/test_threshold_layer.cpp b/src/caffe/test/test_threshold_layer.cpp index 8303e44..a34972e 100644 --- a/src/caffe/test/test_threshold_layer.cpp +++ b/src/caffe/test/test_threshold_layer.cpp @@ -99,52 +99,52 @@ TYPED_TEST(ThresholdLayerTest, TestCPU2) { } } -TYPED_TEST(ThresholdLayerTest, TestGPU) { - LayerParameter layer_param; - Caffe::set_mode(Caffe::GPU); - ThresholdLayer layer(layer_param); - layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_)); - layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_)); - // Now, check values - const TypeParam* bottom_data = this->blob_bottom_->cpu_data(); - const TypeParam* top_data = this->blob_top_->cpu_data(); - const TypeParam threshold_ = layer_param.threshold_param().threshold(); - for (int i = 0; i < this->blob_bottom_->count(); ++i) { - EXPECT_GE(top_data[i], 0.); - EXPECT_LE(top_data[i], 1.); - if (top_data[i] == 0) { - EXPECT_LE(bottom_data[i], threshold_); - } - if (top_data[i] == 1) { - EXPECT_GT(bottom_data[i], threshold_); - } - } -} +// TYPED_TEST(ThresholdLayerTest, TestGPU) { +// LayerParameter layer_param; +// Caffe::set_mode(Caffe::GPU); +// ThresholdLayer layer(layer_param); +// layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_)); +// layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_)); +// // Now, check values +// const TypeParam* bottom_data = this->blob_bottom_->cpu_data(); +// const TypeParam* top_data = this->blob_top_->cpu_data(); +// const TypeParam threshold_ = layer_param.threshold_param().threshold(); +// for (int i = 0; i < this->blob_bottom_->count(); ++i) { +// EXPECT_GE(top_data[i], 0.); +// EXPECT_LE(top_data[i], 1.); +// if (top_data[i] == 0) { +// EXPECT_LE(bottom_data[i], threshold_); +// } +// if (top_data[i] == 1) { +// EXPECT_GT(bottom_data[i], threshold_); +// } +// } +// } -TYPED_TEST(ThresholdLayerTest, TestGPU2) { - LayerParameter layer_param; - Caffe::set_mode(Caffe::GPU); - ThresholdParameter* threshold_param = - layer_param.mutable_threshold_param(); - threshold_param->set_threshold(0.5); - ThresholdLayer layer(layer_param); - layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_)); - layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_)); - // Now, check values - const TypeParam* bottom_data = this->blob_bottom_->cpu_data(); - const TypeParam* top_data = this->blob_top_->cpu_data(); - const TypeParam threshold_ = layer_param.threshold_param().threshold(); - EXPECT_FLOAT_EQ(threshold_, 0.5); - for (int i = 0; i < this->blob_bottom_->count(); ++i) { - EXPECT_GE(top_data[i], 0.); - EXPECT_LE(top_data[i], 1.); - if (top_data[i] == 0) { - EXPECT_LE(bottom_data[i], threshold_); - } - if (top_data[i] == 1) { - EXPECT_GT(bottom_data[i], threshold_); - } - } -} +// TYPED_TEST(ThresholdLayerTest, TestGPU2) { +// LayerParameter layer_param; +// Caffe::set_mode(Caffe::GPU); +// ThresholdParameter* threshold_param = +// layer_param.mutable_threshold_param(); +// threshold_param->set_threshold(0.5); +// ThresholdLayer layer(layer_param); +// layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_)); +// layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_)); +// // Now, check values +// const TypeParam* bottom_data = this->blob_bottom_->cpu_data(); +// const TypeParam* top_data = this->blob_top_->cpu_data(); +// const TypeParam threshold_ = layer_param.threshold_param().threshold(); +// EXPECT_FLOAT_EQ(threshold_, 0.5); +// for (int i = 0; i < this->blob_bottom_->count(); ++i) { +// EXPECT_GE(top_data[i], 0.); +// EXPECT_LE(top_data[i], 1.); +// if (top_data[i] == 0) { +// EXPECT_LE(bottom_data[i], threshold_); +// } +// if (top_data[i] == 1) { +// EXPECT_GT(bottom_data[i], threshold_); +// } +// } +// } } // namespace caffe -- 2.7.4