diff options
author | Eric Tzeng <eric.s.tzeng@gmail.com> | 2014-03-17 13:04:10 -0700 |
---|---|---|
committer | Eric Tzeng <eric.s.tzeng@gmail.com> | 2014-03-17 13:04:10 -0700 |
commit | f6780f982a912ddc118697197da5c7425960c293 (patch) | |
tree | 76b8d6f3610f381a275a73cfa4904fbb995420c2 /src | |
parent | 76e58d5082e7502281ab1b15657e96dbf0519914 (diff) | |
download | caffeonacl-f6780f982a912ddc118697197da5c7425960c293.tar.gz caffeonacl-f6780f982a912ddc118697197da5c7425960c293.tar.bz2 caffeonacl-f6780f982a912ddc118697197da5c7425960c293.zip |
Fix a test assuming CPU mode without explicitly setting it.
Diffstat (limited to 'src')
-rw-r--r-- | src/caffe/test/test_common.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/caffe/test/test_common.cpp b/src/caffe/test/test_common.cpp index 8b521cc7..275c6e1b 100644 --- a/src/caffe/test/test_common.cpp +++ b/src/caffe/test/test_common.cpp @@ -24,6 +24,7 @@ TEST_F(CommonTest, TestVslStream) { } TEST_F(CommonTest, TestBrewMode) { + Caffe::set_mode(Caffe::CPU); EXPECT_EQ(Caffe::mode(), Caffe::CPU); Caffe::set_mode(Caffe::GPU); EXPECT_EQ(Caffe::mode(), Caffe::GPU); |