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 8b521cc..275c6e1 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);