You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by wa...@apache.org on 2016/12/09 13:44:59 UTC
incubator-singa git commit: SINGA-279 Unit Test test_layer failed
when CUDNN version is 4.0
Repository: incubator-singa
Updated Branches:
refs/heads/master 5953e7bfa -> ee93b7dda
SINGA-279 Unit Test test_layer failed when CUDNN version is 4.0
Fixed a bug by removing dropout layer from test_layer unit test when
CUDNN version is 4.0
Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/ee93b7dd
Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/ee93b7dd
Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/ee93b7dd
Branch: refs/heads/master
Commit: ee93b7dda1aa455d5d33807756d592f8f878dc65
Parents: 5953e7b
Author: WANG Ji <ij...@gmail.com>
Authored: Fri Dec 9 21:30:58 2016 +0800
Committer: WANG Ji <ij...@gmail.com>
Committed: Fri Dec 9 21:30:58 2016 +0800
----------------------------------------------------------------------
test/singa/test_layer.cc | 11 +++++------
1 file changed, 5 insertions(+), 6 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/ee93b7dd/test/singa/test_layer.cc
----------------------------------------------------------------------
diff --git a/test/singa/test_layer.cc b/test/singa/test_layer.cc
index 0bc4b12..64c94d5 100644
--- a/test/singa/test_layer.cc
+++ b/test/singa/test_layer.cc
@@ -21,9 +21,9 @@
#include "singa/singa_config.h"
TEST(Layer, CreateLayer) {
- std::vector<std::string> types{
- "convolution", "dense", "dropout", "relu", "batchnorm",
- "flatten", "lrn", "pooling", "prelu", "softmax"};
+ std::vector<std::string> types{"convolution", "dense", "dropout", "relu",
+ "batchnorm", "flatten", "lrn", "pooling",
+ "prelu", "softmax"};
for (auto type : types) {
auto layer = singa::CreateLayer("singacpp_" + type);
// EXPECT_EQ(layer->layer_type(), type);
@@ -32,9 +32,8 @@ TEST(Layer, CreateLayer) {
#ifdef USE_CUDNN
TEST(Layer, CreateCudnnLayer) {
- std::vector<std::string> types{
- "convolution", "dropout", "relu", "batchnorm",
- "lrn", "pooling", "softmax"};
+ std::vector<std::string> types{"convolution", "relu", "batchnorm",
+ "lrn", "pooling", "softmax"};
#if CUDNN_MAJOR >= 5
types.push_back("dropout");
#endif