singa-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From wang...@apache.org
Subject incubator-singa git commit: SINGA-261 Add version ID into the checkpoint files
Date Tue, 06 Dec 2016 14:23:40 GMT
Repository: incubator-singa
Updated Branches:
  refs/heads/master 097126855 -> 5953e7bfa


SINGA-261 Add version ID into the checkpoint files

Fixed a bug from using 'CUDNN_VERSION_MAJOR', which should be 'CUDNN_MAJOR'


Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/5953e7bf
Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/5953e7bf
Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/5953e7bf

Branch: refs/heads/master
Commit: 5953e7bfaeff97518e5259fdc806a78718b89f35
Parents: 0971268
Author: wangwei <wangwei@comp.nus.edu.sg>
Authored: Tue Dec 6 22:20:14 2016 +0800
Committer: wangwei <wangwei@comp.nus.edu.sg>
Committed: Tue Dec 6 22:20:14 2016 +0800

----------------------------------------------------------------------
 test/singa/test_layer.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/5953e7bf/test/singa/test_layer.cc
----------------------------------------------------------------------
diff --git a/test/singa/test_layer.cc b/test/singa/test_layer.cc
index 7726a4a..0bc4b12 100644
--- a/test/singa/test_layer.cc
+++ b/test/singa/test_layer.cc
@@ -35,7 +35,7 @@ TEST(Layer, CreateCudnnLayer) {
   std::vector<std::string> types{
       "convolution", "dropout", "relu", "batchnorm",
       "lrn",   "pooling", "softmax"};
-#if CUDNN_VERSION_MAJOR >= 5
+#if CUDNN_MAJOR >= 5
   types.push_back("dropout");
 #endif
   for (auto type : types) {


Mime
View raw message