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 <[email protected]> Authored: Tue Dec 6 22:20:14 2016 +0800 Committer: wangwei <[email protected]> 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) {
