From 1172df968dcf2b75f89c8a69c6902106ddd9e4e5 Mon Sep 17 00:00:00 2001 From: Wangzheee <634486483@qq.com> Date: Thu, 22 Dec 2022 20:05:35 +0800 Subject: [PATCH] fix int64 (#49265) --- .../analysis/ir_passes/tensorrt_subgraph_pass.cc | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc b/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc index d3a0b4ede4..84eac39a31 100644 --- a/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc +++ b/paddle/fluid/inference/analysis/ir_passes/tensorrt_subgraph_pass.cc @@ -275,10 +275,13 @@ void TensorRtSubgraphPass::CreateTensorRTOp( << "tensorrt_subgraph's input named " << tmp_name << " having int64 dtype in pdmodel description, we will cast them to " "int32 dtype to feed them into paddle-trt."; - PADDLE_ENFORCE_EQ(scope->FindVar(tmp_name), - nullptr, - platform::errors::InvalidArgument( - "The var name %s has exists in scope.", tmp_name)); + /* + PADDLE_ENFORCE_EQ(scope->FindVar(tmp_name), + nullptr, + platform::errors::InvalidArgument( + "The var name %s has exists in scope.", + tmp_name)); + */ scope->Var(tmp_name); } } -- GitLab