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 <[email protected]> Authored: Fri Dec 9 21:30:58 2016 +0800 Committer: WANG Ji <[email protected]> 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
