From 97104695c161ba9ebec6cb2b6df60d51b025325d Mon Sep 17 00:00:00 2001
From: baoachun <962571062@qq.com>
Date: Thu, 21 Apr 2022 15:11:54 +0800
Subject: [PATCH] update gpu fp16 op blacklist (#41703) (#42051)

* 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 bdfe0e46e9c..6bfc16d7d0c 100644
--- a/paddle/fluid/inference/api/paddle_analysis_config.h
+++ b/paddle/fluid/inference/api/paddle_analysis_config.h
@@ -874,7 +874,11 @@ struct PD_INFER_DECL AnalysisConfig {
   bool thread_local_stream_{false};
   bool use_gpu_fp16_{false};
   std::unordered_set<std::string> 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