Set dev

7 years ago · 99f85a9fbc
parent d26f093f9d
commit 99f85a9fbc
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@ -131,6 +131,7 @@ struct ScaleLossGradOpHandle : public OpHandle {
      : coeff_(static_cast<float>(1.0 / num_dev)),
        scope_(scope),
        place_(place) {
+    cudaSetDevice(boost::get<platform::CUDAPlace>(place_).device);
    PADDLE_ENFORCE(cudaEventCreateWithFlags(&ev_, cudaEventDisableTiming));
    VLOG(3) << "Create " << ev_;
  }