未验证 提交 55096a1c 编写于 作者: B baoachun 提交者: GitHub

update gpu fp16 op blacklist (#41703)

* update gpu fp16 op blacklist

* update blacklist
上级 6da637e8
...@@ -888,7 +888,11 @@ struct PD_INFER_DECL AnalysisConfig { ...@@ -888,7 +888,11 @@ struct PD_INFER_DECL AnalysisConfig {
bool thread_local_stream_{false}; bool thread_local_stream_{false};
bool use_gpu_fp16_{false}; bool use_gpu_fp16_{false};
std::unordered_set<std::string> gpu_fp16_disabled_op_types_{ 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}; bool use_cudnn_{false};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册