Skip to content

Commit 1c43ef4

Browse files
author
chengduozh
committed
fix GetTensorFromVar
test=release/1.1
1 parent 69c2acb commit 1c43ef4

File tree

3 files changed

+6
-8
lines changed

3 files changed

+6
-8
lines changed

paddle/fluid/framework/operator.cc

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -358,11 +358,11 @@ static bool VarIsTensor(const Variable* var) {
358358
return var->IsType<LoDTensor>() || var->IsType<SelectedRows>();
359359
}
360360

361-
const Tensor* GetTensorFromVar(Variable* var) {
361+
const Tensor* GetTensorFromVar(const Variable* var) {
362362
if (var->IsType<LoDTensor>()) {
363-
return var->GetMutable<LoDTensor>();
363+
return static_cast<const Tensor*>(&(var->Get<LoDTensor>()));
364364
} else if (var->IsType<SelectedRows>()) {
365-
return var->GetMutable<SelectedRows>()->mutable_value();
365+
return &(var->Get<SelectedRows>().value());
366366
} else {
367367
PADDLE_THROW("Variable type_id %s, expect LoDTensor/SelectedRows.",
368368
var->Type().name());
@@ -415,8 +415,7 @@ bool ExecutionContext::HasOutput(const std::string& name) const {
415415
template <>
416416
const Tensor* ExecutionContext::Input<Tensor>(const std::string& name) const {
417417
auto* var = InputVar(name);
418-
return var == nullptr ? nullptr
419-
: GetTensorFromVar(const_cast<Variable*>(var));
418+
return var == nullptr ? nullptr : GetTensorFromVar(var);
420419
}
421420

422421
template <>

paddle/fluid/framework/operator.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ inline std::string GradVarName(const std::string& var_name) {
6363
}
6464

6565
proto::VarType::Type GetDataTypeOfVar(const Variable* var);
66-
const Tensor* GetTensorFromVar(Variable* var);
66+
const Tensor* GetTensorFromVar(const Variable* var);
6767

6868
class OperatorBase;
6969
class ExecutionContext;

paddle/fluid/operators/sum_op.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,8 +83,7 @@ class SumOp : public framework::OperatorWithKernel {
8383
int dtype = -1;
8484
for (auto& x_var : x_vars) {
8585
// FIXME(zcd): The input x_var may be SelectedRows or LoDTensor.
86-
auto tensor = framework::GetTensorFromVar(
87-
const_cast<framework::Variable*>(x_var));
86+
auto tensor = framework::GetTensorFromVar(x_var);
8887
if (tensor->numel() == 0) {
8988
continue;
9089
}

0 commit comments

Comments
 (0)