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/06 14:23:40 UTC
incubator-singa git commit: SINGA-261 Add version ID into the
checkpoint files
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 <wa...@comp.nus.edu.sg>
Authored: Tue Dec 6 22:20:14 2016 +0800
Committer: wangwei <wa...@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) {