figured out the bug - it's curand

[jacinto-ai/caffe-jacinto.git] / src / caffeine / test / test_neuron_layer.cpp

diff --git a/src/caffeine/test/test_neuron_layer.cpp b/src/caffeine/test/test_neuron_layer.cpp
index 06d48529856cdbaa2392c2d0cfb3196b18d700f2..45c52c3abfbb201a9cf9615fb94e0e548ce36bc0 100644 (file)
--- a/src/caffeine/test/test_neuron_layer.cpp
+++ b/src/caffeine/test/test_neuron_layer.cpp
@@ -148,9 +148,7 @@ TYPED_TEST(NeuronLayerTest, TestDropoutGPU) {
   }
 }
 
-/*
- * Yangqing's note: disabled due to some curand problem.
- *
+
 TYPED_TEST(NeuronLayerTest, TestDropoutGradientGPU) {
   LayerParameter layer_param;
   Caffeine::set_mode(Caffeine::GPU);
@@ -158,7 +156,6 @@ TYPED_TEST(NeuronLayerTest, TestDropoutGradientGPU) {
   GradientChecker<TypeParam> checker(1e-2, 1e-3);
   checker.CheckGradient(layer, this->blob_bottom_vec_, this->blob_top_vec_);
 }
-*/
 
 
 TYPED_TEST(NeuronLayerTest, TestDropoutGPUTestPhase) {