executor.cc 2.6 KB
Newer Older
Q
qijun 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */

#include "paddle/framework/executor.h"
Y
Yang Yang 已提交
16

Y
Yang Yang 已提交
17
#include <algorithm>
Y
Yang Yang 已提交
18
#include <iostream>
Q
qijun 已提交
19
#include <memory>
Y
Yang Yang 已提交
20
#include <set>
Y
Yang Yang 已提交
21
#include <vector>
Y
Yang Yang 已提交
22

Y
Yang Yang 已提交
23
#include "paddle/framework/lod_tensor.h"
Q
qijun 已提交
24
#include "paddle/framework/op_registry.h"
Q
qijun 已提交
25
#include "paddle/framework/scope.h"
Q
qijun 已提交
26 27 28 29

namespace paddle {
namespace framework {

Y
Yang Yang 已提交
30 31 32
const std::string kFeedOpType = "feed";
const std::string kFetchOpType = "fetch";

Q
qijun 已提交
33
Executor::Executor(const std::vector<platform::Place>& places) {
Y
Yang Yang 已提交
34
  PADDLE_ENFORCE_GT(places.size(), 0);
Q
qijun 已提交
35
  device_contexts_.resize(places.size());
Q
qijun 已提交
36
  for (size_t i = 0; i < places.size(); i++) {
Q
qijun 已提交
37
    if (platform::is_cpu_place(places[i])) {
Q
qijun 已提交
38 39 40
      device_contexts_[i] = new platform::CPUDeviceContext(
          boost::get<platform::CPUPlace>(places[i]));
    } else if (platform::is_gpu_place(places[i])) {
Q
qijun 已提交
41
#ifdef PADDLE_WITH_CUDA
Q
qijun 已提交
42 43
      device_contexts_[i] = new platform::CUDADeviceContext(
          boost::get<platform::GPUPlace>(places[i]));
Q
qijun 已提交
44
#else
Q
qijun 已提交
45 46 47
      PADDLE_THROW(
          "'GPUPlace' is not supported, Please re-compile with WITH_GPU "
          "option");
Q
qijun 已提交
48 49
#endif
    }
Q
qijun 已提交
50 51 52
  }
}

Q
qijun 已提交
53 54
Executor::~Executor() {
  for (auto& device_context : device_contexts_) {
Y
Yang Yang 已提交
55
    delete device_context;
Q
qijun 已提交
56 57 58
  }
}

Y
Yang Yang 已提交
59
void Executor::Run(const ProgramDesc& pdesc, Scope* scope, int block_id) {
Y
Yang Yang 已提交
60
  // TODO(tonyyang-svail):
Y
Yang Yang 已提交
61
  //    - only runs on the first device (i.e. no interdevice communication)
Y
Yang Yang 已提交
62
  //    - will change to use multiple blocks for RNN op and Cond Op
Y
Yang Yang 已提交
63 64
  PADDLE_ENFORCE_GT(pdesc.blocks_size(), block_id);
  auto& block = pdesc.blocks(block_id);
Y
Yang Yang 已提交
65
  auto& device = device_contexts_[0];
Y
Yang Yang 已提交
66

Y
Yang Yang 已提交
67 68
  Scope& local_scope = scope->NewScope();

Y
Yang Yang 已提交
69 70 71 72 73
  for (auto& var : block.vars()) {
    if (var.persistable()) {
      scope->Var(var.name());
    } else {
      local_scope.Var(var.name());
Y
Yang Yang 已提交
74
    }
Y
Yang Yang 已提交
75
  }
Y
Yang Yang 已提交
76

Y
Yang Yang 已提交
77
  for (auto& op_desc : block.ops()) {
78 79
    auto op = paddle::framework::OpRegistry::CreateOp(
        op_desc, const_cast<ProgramDesc*>(&pdesc));
Y
Yang Yang 已提交
80
    op->Run(local_scope, *device);
Q
qijun 已提交
81
  }
Y
Yang Yang 已提交
82

Y
Yang Yang 已提交
83
  // TODO(tonyyang-svail):
Y
Yang Yang 已提交
84
  //  - Destroy local_scope
Q
qijun 已提交
85 86 87 88
}

}  // namespace framework
}  // namespace paddle