Skip to content

Commit 8c01af2

Browse files
committed
update
1 parent b2acdbb commit 8c01af2

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/framework/operator.cc

+3-3
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,9 @@ void UseALL() {
7474

7575
static DDim GetDims(const Scope& scope, const std::string& name) {
7676
Variable* var = scope.FindVar(name);
77-
if (var->IsType<LoDTensor>()) {
77+
if (var && var->IsType<LoDTensor>()) {
7878
return var->Get<LoDTensor>().dims();
79-
} else if (var->IsType<SelectedRows>()) {
79+
} else if (var && var->IsType<SelectedRows>()) {
8080
return var->Get<SelectedRows>().GetCompleteDims();
8181
} else {
8282
return DDim({-1});
@@ -123,7 +123,7 @@ std::string OperatorBase::DebugStringEx(const Scope* scope) const {
123123
ss << input.first << "[";
124124
for (size_t i = 0; i < input.second.size(); ++i) {
125125
ss << input.second[i];
126-
if (scope && scope->FindVar(input.second[i])) {
126+
if (scope) {
127127
ss << "(" << GetDims(*scope, input.second[i]) << ")";
128128
}
129129
if (i != input.second.size() - 1) {

0 commit comments

Comments
 (0)