From 078c094f486365e4b3c8dbb45486de17d9828124 Mon Sep 17 00:00:00 2001 From: Yangqing Jia Date: Tue, 17 Sep 2013 17:39:47 -0700 Subject: [PATCH] figured out the bug - it's curand --- src/caffeine/test/test_neuron_layer.cpp | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/caffeine/test/test_neuron_layer.cpp b/src/caffeine/test/test_neuron_layer.cpp index 06d48529..45c52c3a 100644 --- 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 checker(1e-2, 1e-3); checker.CheckGradient(layer, this->blob_bottom_vec_, this->blob_top_vec_); } -*/ TYPED_TEST(NeuronLayerTest, TestDropoutGPUTestPhase) { -- 2.39.2