Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2302
Star 20931
Fork 5422
P
Paddle
未验证 提交 ac666538 编写于 作者: F feng_shuai 提交者： GitHub
浏览文件
操作

fix:use constant_fold for vit pass (#47211)

上级 ec5b27f5
显示空白变更内容
内联 并排
Showing with 3 addition and 3 deletion
paddle/fluid/inference/api/paddle_pass_builder.cc
浏览文件 @ ac666538
...@@ -106,6 +106,7 @@ const std::vector<std::string> kTRTSubgraphPasses({ ...@@ -106,6 +106,7 @@ const std::vector<std::string> kTRTSubgraphPasses({
"delete_c_identity_op_pass", // "delete_c_identity_op_pass", //
"trt_multihead_matmul_fuse_pass_v2", // "trt_multihead_matmul_fuse_pass_v2", //
"trt_multihead_matmul_fuse_pass_v3", // "trt_multihead_matmul_fuse_pass_v3", //
"constant_folding_pass", //
"vit_attention_fuse_pass", // "vit_attention_fuse_pass", //
"trt_skip_layernorm_fuse_pass", // "trt_skip_layernorm_fuse_pass", //
"preln_skip_layernorm_fuse_pass", // "preln_skip_layernorm_fuse_pass", //
...@@ -127,7 +128,6 @@ const std::vector<std::string> kTRTSubgraphPasses({ ...@@ -127,7 +128,6 @@ const std::vector<std::string> kTRTSubgraphPasses({
// "yolo_box_fuse_pass", // // "yolo_box_fuse_pass", //
"dense_fc_to_sparse_pass", // "dense_fc_to_sparse_pass", //
"dense_multihead_matmul_to_sparse_pass", // "dense_multihead_matmul_to_sparse_pass", //
"constant_folding_pass",
"tensorrt_subgraph_pass", // "tensorrt_subgraph_pass", //
"conv_bn_fuse_pass", // "conv_bn_fuse_pass", //
#if CUDNN_VERSION >= 7100 // To run conv_fusion, the version of cudnn must be #if CUDNN_VERSION >= 7100 // To run conv_fusion, the version of cudnn must be
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或