未验证 提交 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), PADDLE_ENFORCE_EQ(scope->FindVar(tmp_name),
nullptr, nullptr,
platform::errors::InvalidArgument( platform::errors::InvalidArgument(
"The var name %s has exists in scope.", tmp_name)); "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.
先完成此消息的编辑!
想要评论请 注册