figured out the bug - it's curand
authorYangqing Jia <jiayq84@gmail.com>
Wed, 18 Sep 2013 00:39:47 +0000 (17:39 -0700)
committerYangqing Jia <jiayq84@gmail.com>
Wed, 18 Sep 2013 00:39:47 +0000 (17:39 -0700)
src/caffeine/test/test_neuron_layer.cpp

index 06d48529856cdbaa2392c2d0cfb3196b18d700f2..45c52c3abfbb201a9cf9615fb94e0e548ce36bc0 100644 (file)
@@ -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) {