correct macro

8 years ago · 3014f6a113
parent e02cc571cf
commit 3014f6a113
2 changed files with 2 additions and 2 deletions
--- a/paddle/framework/executor.cc
+++ b/paddle/framework/executor.cc
@ -30,7 +30,7 @@ Executor::Executor(const std::vector<platform::Place>& places) {
      device_contexts_[i] = new platform::CPUDeviceContext(
          boost::get<platform::CPUPlace>(places[i]));
    } else if (platform::is_gpu_place(places[i])) {
-#ifndef PADDLE_ONLY_CPU
+#ifdef PADDLE_WITH_GPU
      device_contexts_[i] = new platform::CUDADeviceContext(
          boost::get<platform::GPUPlace>(places[i]));
 #else
--- a/paddle/framework/executor_test.cc
+++ b/paddle/framework/executor_test.cc
@ -103,7 +103,7 @@ TEST_F(ExecutorTester, InitCPU) {
  delete executor;
 }

-#ifndef PADDLE_ONLY_CPU
+#ifdef PADDLE_WITH_GPU
 TEST_F(ExecutorTester, InitGPU) {
  std::vector<Place> places;
  GPUPlace gpu_place0(0);