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 d25f51e4fd4..9c48d822b4d 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<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