未验证 提交 766a4ca9 编写于 作者: X xiaoxiaohehe001 提交者: GitHub

[Paddle Inference] Fix range fp32 input below trt 8.4. (#50221)

上级 1dedaada
......@@ -360,6 +360,15 @@ struct SimpleOpTypeSetTeller : public Teller {
if (!with_dynamic_shape) {
return false;
}
#if IS_TRT_VERSION_LT(8400)
auto* block = desc.Block();
auto start_var_name = desc.Input("Start")[0];
auto* start_var_desc = block->FindVar(start_var_name);
auto start_dtype = start_var_desc->GetDataType();
if (start_dtype == framework::proto::VarType::FP32) {
return false;
}
#endif
}
if (op_type == "sign") {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册