Another bugfix related to my CPU/GPU test changes: make NetTest a

MultiDeviceTest (Caffe:set_mode(Caffe::CPU/GPU) isn't run without this).

diff --git a/src/caffe/test/test_net.cpp b/src/caffe/test/test_net.cpp
index cd6daeb..c30f416 100644 (file)
--- a/src/caffe/test/test_net.cpp
+++ b/src/caffe/test/test_net.cpp
@@ -19,7 +19,7 @@ using std::ostringstream;
 namespace caffe {
 
 template <typename TypeParam>
-class NetTest : public ::testing::Test {
+class NetTest : public MultiDeviceTest<TypeParam> {
   typedef typename TypeParam::Dtype Dtype;
 
  protected: