From fe00a6cf97ebc128549d7f6a30f9659718ca6fe2 Mon Sep 17 00:00:00 2001
From: Haihao Shen <haihao.shen@intel.com>
Date: Wed, 5 Dec 2018 08:42:10 +0800
Subject: [PATCH] Combine the key for src reorder to reduce malloc

---
 paddle/fluid/operators/conv_mkldnn_op.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/conv_mkldnn_op.cc b/paddle/fluid/operators/conv_mkldnn_op.cc
index fb55e8e5b24..2be5a27d2c5 100644
--- a/paddle/fluid/operators/conv_mkldnn_op.cc
+++ b/paddle/fluid/operators/conv_mkldnn_op.cc
@@ -139,7 +139,7 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
     auto dst_key = key + "@dst_mem_p";
     auto src_key = key + "@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));
     if(conv_p == nullptr){
       if(is_INT8){
-- 
GitLab