Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 2 年 前同步成功

通知 2325
Star 20933
Fork 5424
P
Paddle
提交 fe00a6cf 编写于 作者: H Haihao Shen
浏览文件
操作

Combine the key for src reorder to reduce malloc

上级 1f4f5c5d
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
paddle/fluid/operators/conv_mkldnn_op.cc
浏览文件 @ fe00a6cf
...@@ -139,7 +139,7 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> { ...@@ -139,7 +139,7 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
auto dst_key = key + "@dst_mem_p"; auto dst_key = key + "@dst_mem_p";
auto src_key = key + "@src_mem_p"; auto src_key = key + "@src_mem_p";
auto user_src_key = key + "@user_src_mem_p"; auto user_src_key = key + "@user_src_mem_p";
auto src_reorder_key = key + "@src_mem_p" + "reorder_p"; auto src_reorder_key = key + "@src_mem_p@reorder_p";
conv_p = std::static_pointer_cast<mkldnn::convolution_forward>(dev_ctx.GetBlob(prim_key)); conv_p = std::static_pointer_cast<mkldnn::convolution_forward>(dev_ctx.GetBlob(prim_key));
if(conv_p == nullptr){ if(conv_p == nullptr){
if(is_INT8){ if(is_INT8){
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或