Fixed CMake

BVLC · Oct 17, 2014 · f5af8fb · f5af8fb
1 parent 2c9abcb
commit f5af8fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/caffe/test/test_imports.cpp b/src/caffe/test/test_imports.cpp
@@ -30,6 +30,7 @@ class ImportsTest : public MultiDeviceTest<TypeParam> {
   }
 
   virtual void InitNet() {
+    string file = CMAKE_SOURCE_DIR "caffe/test/test_data/module.prototxt";
     string proto =
         "name: 'TestNetwork' "
         "layers: { "
@@ -56,7 +57,7 @@ class ImportsTest : public MultiDeviceTest<TypeParam> {
         "  name: 'import' "
         "  type: IMPORT "
         "  import_param { "
-        "    net: 'src/caffe/test/test_data/module.prototxt' "
+        "    net: '" + file + "' "
         "    var { name: 'num_output' value: '1000' } "
         "  } "
         "} "