From 55096a1c602230b9f93bb8941c611758bf39e2e0 Mon Sep 17 00:00:00 2001 From: baoachun <962571062@qq.com> Date: Tue, 19 Apr 2022 15:03:31 +0800 Subject: [PATCH] update gpu fp16 op blacklist (#41703) * update gpu fp16 op blacklist * update blacklist --- paddle/fluid/inference/api/paddle_analysis_config.h | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/inference/api/paddle_analysis_config.h b/paddle/fluid/inference/api/paddle_analysis_config.h index d25f51e4fd..9c48d822b4 100644 --- a/paddle/fluid/inference/api/paddle_analysis_config.h +++ b/paddle/fluid/inference/api/paddle_analysis_config.h @@ -888,7 +888,11 @@ struct PD_INFER_DECL AnalysisConfig { bool thread_local_stream_{false}; bool use_gpu_fp16_{false}; std::unordered_set gpu_fp16_disabled_op_types_{ - "conv2d_fusion", "conv2d", "roll", "strided_slice"}; + "conv2d_fusion", "conv2d", "roll", "strided_slice", "depthwise_conv2d", + "unfold", "generate_proposals_v2", "nearest_interp_v2", + "bilinear_interp_v2" + "yolo_box", + "multiclass_nms3", "matrix_nms"}; bool use_cudnn_{false}; -- GitLab