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

update gpu fp16 op blacklist (#41703) (#42051)

* update gpu fp16 op blacklist

* update blacklist
上级 04f20b83
......@@ -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};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册