Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2297
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 d6e15914 编写于 作者: 王 王明冬 提交者： GitHub
浏览文件
操作

[Infrt] fix ci bug. test=document_fix (#41663)

上级 b45f80dd
隐藏空白更改
内联 并排
Showing with 2 addition and 0 deletion
paddle/infrt/api/infrt_api.cc
浏览文件 @ d6e15914
...@@ -257,11 +257,13 @@ int InfRtPredictor::Init(const InfRtConfig& config) { ...@@ -257,11 +257,13 @@ int InfRtPredictor::Init(const InfRtConfig& config) {
::mlir::OpPassManager& pass_manager = pm.nest<::mlir::FuncOp>(); ::mlir::OpPassManager& pass_manager = pm.nest<::mlir::FuncOp>();
if (config.tensorrt_enabled()) { if (config.tensorrt_enabled()) {
pass_manager.addPass(::infrt::CreateInfrtWeightsUnfoldPass()); pass_manager.addPass(::infrt::CreateInfrtWeightsUnfoldPass());
#if defined(INFRT_WITH_GPU) && defined(INFRT_WITH_TRT)
pass_manager.addPass(::infrt::trt::CreateTrtOpTellerPass()); pass_manager.addPass(::infrt::trt::CreateTrtOpTellerPass());
pass_manager.addPass(::infrt::trt::CreateTrtGraphFusePass()); pass_manager.addPass(::infrt::trt::CreateTrtGraphFusePass());
pass_manager.addPass(::infrt::trt::CreateTrtGraphSplitPass(1)); pass_manager.addPass(::infrt::trt::CreateTrtGraphSplitPass(1));
pass_manager.addPass(::infrt::trt::CreateTrtOpConverterPass()); pass_manager.addPass(::infrt::trt::CreateTrtOpConverterPass());
pass_manager.addPass(::infrt::trt::CreateTrtTypeConvertPass()); pass_manager.addPass(::infrt::trt::CreateTrtTypeConvertPass());
#endif
pass_manager.addPass(::mlir::createCanonicalizerPass()); pass_manager.addPass(::mlir::createCanonicalizerPass());
} else { } else {
std::vector<::infrt::Place> valid_places = { std::vector<::infrt::Place> valid_places = {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或