未验证 提交 1172df96 编写于 作者: W Wangzheee 提交者: GitHub

fix int64 (#49265)

上级 80f5e25e
...@@ -275,10 +275,13 @@ void TensorRtSubgraphPass::CreateTensorRTOp( ...@@ -275,10 +275,13 @@ void TensorRtSubgraphPass::CreateTensorRTOp(
<< "tensorrt_subgraph's input named " << tmp_name << "tensorrt_subgraph's input named " << tmp_name
<< " having int64 dtype in pdmodel description, we will cast them to " << " having int64 dtype in pdmodel description, we will cast them to "
"int32 dtype to feed them into paddle-trt."; "int32 dtype to feed them into paddle-trt.";
PADDLE_ENFORCE_EQ(scope->FindVar(tmp_name), /*
nullptr, PADDLE_ENFORCE_EQ(scope->FindVar(tmp_name),
platform::errors::InvalidArgument( nullptr,
"The var name %s has exists in scope.", tmp_name)); platform::errors::InvalidArgument(
"The var name %s has exists in scope.",
tmp_name));
*/
scope->Var(tmp_name); scope->Var(tmp_name);
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册