fix

7 years ago · ce55975bb5
parent 18ac6947d0
commit ce55975bb5
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/framework/init.cc
+++ b/paddle/fluid/framework/init.cc
@ -44,6 +44,7 @@ void InitGflags(std::vector<std::string> &argv) {
 }

 void InitP2P(int count) {
+#ifdef PADDLE_WITH_CUDA
  std::call_once(p2p_init_flag, [&]() {
    for (int i = 0; i < count; ++i) {
      for (int j = 0; j < count; ++j) {
@ -60,6 +61,7 @@ void InitP2P(int count) {
      }
    }
  });
+#endif
 }

 void InitDevices() {