singa-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From wan...@apache.org
Subject incubator-singa git commit: SINGA-279 Unit Test test_layer failed when CUDNN version is 4.0
Date Fri, 09 Dec 2016 13:44:59 GMT
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 <ijingobravo@gmail.com>
Authored: Fri Dec 9 21:30:58 2016 +0800
Committer: WANG Ji <ijingobravo@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


Mime
View raw message