From e9ca7600a00f2389c5e5f53fbce29c4f660596d9 Mon Sep 17 00:00:00 2001 From: feng_shuai Date: Wed, 30 Nov 2022 12:34:24 +0800 Subject: [PATCH] feat:add the support for vit_attention_op on gpu (#48515) --- paddle/fluid/inference/api/paddle_pass_builder.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/inference/api/paddle_pass_builder.cc b/paddle/fluid/inference/api/paddle_pass_builder.cc index f02776d00f..16db8bee9e 100644 --- a/paddle/fluid/inference/api/paddle_pass_builder.cc +++ b/paddle/fluid/inference/api/paddle_pass_builder.cc @@ -216,6 +216,7 @@ GpuPassStrategy::GpuPassStrategy() : PassStrategy({}) { "conv_eltwiseadd_bn_fuse_pass", // "embedding_eltwise_layernorm_fuse_pass", // "multihead_matmul_fuse_pass_v2", // + "vit_attention_fuse_pass", // "fused_multi_transformer_encoder_pass", // "fused_multi_transformer_decoder_pass", // "fused_multi_transformer_encoder_fuse_qkv_pass", // -- GitLab