test=develop
for_weibo
Xin Pan 7 years ago
parent fb8ae30331
commit 70981f5d79

@ -143,14 +143,12 @@ RuntimeContext::RuntimeContext(const VariableNameMap& innames,
for (auto& var_name_item : innames) { for (auto& var_name_item : innames) {
std::vector<Variable*>& input_vars = inputs[var_name_item.first]; std::vector<Variable*>& input_vars = inputs[var_name_item.first];
for (auto& var_name : var_name_item.second) { for (auto& var_name : var_name_item.second) {
LOG(ERROR) << "first in " << var_name_item.first << ":" << var_name;
input_vars.push_back(scope.FindVar(var_name)); input_vars.push_back(scope.FindVar(var_name));
} }
} }
for (auto& var_name_item : outnames) { for (auto& var_name_item : outnames) {
std::vector<Variable*>& output_vars = outputs[var_name_item.first]; std::vector<Variable*>& output_vars = outputs[var_name_item.first];
for (auto& var_name : var_name_item.second) { for (auto& var_name : var_name_item.second) {
LOG(ERROR) << "first out " << var_name_item.first << ":" << var_name;
output_vars.push_back(scope.FindVar(var_name)); output_vars.push_back(scope.FindVar(var_name));
} }
} }
@ -441,22 +439,13 @@ const Variable* ExecutionContext::InputVar(const std::string& name) const {
return it->second.empty() ? nullptr : it->second[0]; return it->second.empty() ? nullptr : it->second[0];
} }
Variable* ExecutionContext::OutputVar(const std::string& name) const { const Variable* ExecutionContext::LegacyInputVar(
auto opt = op_.Output(name); const std::string& name) const {
return opt == kEmptyVarName ? nullptr : scope_.FindVar(opt); auto ipt = op_.Input(name);
} return ipt == kEmptyVarName ? nullptr : scope_.FindVar(ipt);
const Variable* ExecutionContext::FastInputVar(const std::string& name) const {
auto it = ctx_.inputs.find(name);
if (it == ctx_.inputs.end()) return nullptr;
PADDLE_ENFORCE_LE(it->second.size(), 1UL,
"Operator %s's input %s should contain only one variable.",
op_.Type(), name);
return it->second.empty() ? nullptr : it->second[0];
} }
Variable* ExecutionContext::FastOutputVar(const std::string& name) const { Variable* ExecutionContext::OutputVar(const std::string& name) const {
auto it = ctx_.outputs.find(name); auto it = ctx_.outputs.find(name);
if (it == ctx_.outputs.end()) return nullptr; if (it == ctx_.outputs.end()) return nullptr;
@ -466,15 +455,20 @@ Variable* ExecutionContext::FastOutputVar(const std::string& name) const {
return it->second.empty() ? nullptr : it->second[0]; return it->second.empty() ? nullptr : it->second[0];
} }
Variable* ExecutionContext::LegacyOutputVar(const std::string& name) const {
auto opt = op_.Output(name);
return opt == kEmptyVarName ? nullptr : scope_.FindVar(opt);
}
template <> template <>
const Tensor* ExecutionContext::Input<Tensor>(const std::string& name) const { const Tensor* ExecutionContext::Input<Tensor>(const std::string& name) const {
return Input<LoDTensor>(name); return Input<LoDTensor>(name);
} }
template <> template <>
const Tensor* ExecutionContext::FastInput<Tensor>( const Tensor* ExecutionContext::LegacyInput<Tensor>(
const std::string& name) const { const std::string& name) const {
return FastInput<LoDTensor>(name); return LegacyInput<LoDTensor>(name);
} }
template <> template <>
@ -502,8 +496,8 @@ Tensor* ExecutionContext::Output<Tensor>(const std::string& name) const {
} }
template <> template <>
Tensor* ExecutionContext::FastOutput<Tensor>(const std::string& name) const { Tensor* ExecutionContext::LegacyOutput<Tensor>(const std::string& name) const {
return FastOutput<LoDTensor>(name); return LegacyOutput<LoDTensor>(name);
} }
template <> template <>
@ -870,7 +864,6 @@ Scope* OperatorWithKernel::PrepareData(
auto& var_name = var_name_item.second[i]; auto& var_name = var_name_item.second[i];
auto* var = scope.FindVar(var_name); auto* var = scope.FindVar(var_name);
input_vars[i] = var; input_vars[i] = var;
LOG(ERROR) << "second in " << var_name_item.first << ":" << var_name;
// Only tensor can be tranfer to another device. // Only tensor can be tranfer to another device.
if (var == nullptr || !VarIsTensor(*var)) { if (var == nullptr || !VarIsTensor(*var)) {
@ -931,7 +924,6 @@ Scope* OperatorWithKernel::PrepareData(
for (size_t i = 0; i < var_name_item.second.size(); ++i) { for (size_t i = 0; i < var_name_item.second.size(); ++i) {
auto& var_name = var_name_item.second[i]; auto& var_name = var_name_item.second[i];
output_vars[i] = scope.FindVar(var_name); output_vars[i] = scope.FindVar(var_name);
LOG(ERROR) << "second out " << var_name_item.first << ":" << var_name;
} }
} }

@ -233,20 +233,20 @@ class ExecutionContext {
} }
template <typename T> template <typename T>
const T* FastInput(const std::string& name) const { const T* LegacyInput(const std::string& name) const {
auto* var = FastInputVar(name); auto* var = LegacyInputVar(name);
return var == nullptr ? nullptr : &var->Get<T>(); return var == nullptr ? nullptr : &var->Get<T>();
} }
template <typename T> template <typename T>
T* FastOutput(const std::string& name) const { T* LegacyOutput(const std::string& name) const {
auto var = FastOutputVar(name); auto var = LegacyOutputVar(name);
return var == nullptr ? nullptr : var->GetMutable<T>(); return var == nullptr ? nullptr : var->GetMutable<T>();
} }
const Variable* FastInputVar(const std::string& name) const; const Variable* LegacyInputVar(const std::string& name) const;
Variable* FastOutputVar(const std::string& name) const; Variable* LegacyOutputVar(const std::string& name) const;
template <typename T> template <typename T>
const std::vector<const T*> MultiInput(const std::string& name) const { const std::vector<const T*> MultiInput(const std::string& name) const {
@ -314,7 +314,7 @@ template <>
const Tensor* ExecutionContext::Input<Tensor>(const std::string& name) const; const Tensor* ExecutionContext::Input<Tensor>(const std::string& name) const;
template <> template <>
const Tensor* ExecutionContext::FastInput<Tensor>( const Tensor* ExecutionContext::LegacyInput<Tensor>(
const std::string& name) const; const std::string& name) const;
template <> template <>
@ -325,7 +325,7 @@ template <>
Tensor* ExecutionContext::Output<Tensor>(const std::string& name) const; Tensor* ExecutionContext::Output<Tensor>(const std::string& name) const;
template <> template <>
Tensor* ExecutionContext::FastOutput<Tensor>(const std::string& name) const; Tensor* ExecutionContext::LegacyOutput<Tensor>(const std::string& name) const;
template <> template <>
std::vector<Tensor*> ExecutionContext::MultiOutput<Tensor>( std::vector<Tensor*> ExecutionContext::MultiOutput<Tensor>(

@ -56,7 +56,7 @@ class PReluOp : public framework::OperatorWithKernel {
protected: protected:
framework::OpKernelType GetExpectedKernelType( framework::OpKernelType GetExpectedKernelType(
const framework::ExecutionContext &ctx) const override { const framework::ExecutionContext &ctx) const override {
return framework::OpKernelType(ctx.FastInput<Tensor>("X")->type(), return framework::OpKernelType(ctx.Input<Tensor>("X")->type(),
ctx.device_context()); ctx.device_context());
} }
}; };

Loading…
Cancel
Save