Set FLAGS_use_gpu in test_DetectionOutput.

8 years ago · b233ed1353
parent 65969dad64
commit b233ed1353
1 changed files with 3 additions and 0 deletions
--- a/paddle/gserver/tests/test_DetectionOutput.cpp
+++ b/paddle/gserver/tests/test_DetectionOutput.cpp
@ -65,9 +65,12 @@ void doOneDetectionOutputTest(MatrixPtr& inputLoc,
  dataLayers[2]->getOutputValue()->copyFrom(*inputConf);

  // test layer initialize
+  bool store_FLAGS_use_gpu = FLAGS_use_gpu;
+  FLAGS_use_gpu = use_gpu;
  std::vector<ParameterPtr> parameters;
  LayerPtr detectionOutputLayer;
  initTestLayer(configt, &layerMap, &parameters, &detectionOutputLayer);
+  FLAGS_use_gpu = store_FLAGS_use_gpu;
  detectionOutputLayer->forward(PASS_GC);
  checkMatrixEqual(detectionOutputLayer->getOutputValue(), result);
 }