projects
/
platform
/
upstream
/
caffeonacl.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
88363e9
)
figured out the bug - it's curand
author
Yangqing Jia
<jiayq84@gmail.com>
Wed, 18 Sep 2013 00:39:47 +0000
(17:39 -0700)
committer
Yangqing Jia
<jiayq84@gmail.com>
Wed, 18 Sep 2013 00:39:47 +0000
(17:39 -0700)
src/caffeine/test/test_neuron_layer.cpp
patch
|
blob
|
history
diff --git
a/src/caffeine/test/test_neuron_layer.cpp
b/src/caffeine/test/test_neuron_layer.cpp
index
06d4852
..
45c52c3
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) {