fix CMAKE

7 years ago · 7ba55aa294
parent 6974265292
commit 7ba55aa294
2 changed files with 3 additions and 3 deletions
--- a/paddle/fluid/train/CMakeLists.txt
+++ b/paddle/fluid/train/CMakeLists.txt
@ -25,4 +25,6 @@ function(train_test TARGET_NAME)
 endfunction(train_test)


-train_test(recognize_digits ARGS mlp conv)
+if(WITH_TESTING)
+  train_test(recognize_digits ARGS mlp conv)
+endif()
--- a/paddle/fluid/train/test_train_recognize_digits.cc
+++ b/paddle/fluid/train/test_train_recognize_digits.cc
@ -52,8 +52,6 @@ void Train() {

  PADDLE_ENFORCE_NE(loss_name, "", "loss not found");

-  // init all parameters
-
  // prepare data
  auto x_var = scope.Var("img");
  auto x_tensor = x_var->GetMutable<framework::LoDTensor>();