Skip to content

Commit 9a81926

Browse files
committed
fix
test=develop
1 parent 6de08b5 commit 9a81926

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

paddle/fluid/framework/op_desc.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,10 @@ class CompileTimeInferShapeContext : public InferShapeContext {
8181
"The %s[%d] is @EMPTY@", out, j);
8282
auto *in_var = block_.FindVarRecursive(Inputs(in)[i]);
8383
auto *out_var = block_.FindVarRecursive(Outputs(out)[j]);
84-
PADDLE_ENFORCE_EQ(in_var->GetType(), proto::VarType::LOD_TENSOR,
85-
"The %d-th output of Output(%s) must be LoDTensor.", j,
86-
out);
87-
84+
if (in_var->GetType() != proto::VarType::LOD_TENSOR) {
85+
VLOG(3) << "input " << in << " is not LodTensor";
86+
return;
87+
}
8888
out_var->SetLoDLevel(in_var->GetLoDLevel());
8989
}
9090

0 commit comments

Comments
 (0)