Fix a test assuming CPU mode without explicitly setting it.

diff --git a/src/caffe/test/test_common.cpp b/src/caffe/test/test_common.cpp
index 8b521cc7ea3cc938d5b6c734e5eba1b95d55ed22..275c6e1bf73d66bb4eb6ba1b91f0c00f92715af0 100644 (file)
--- 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);