提交 26fb34c3 编写于 作者: Y Yu Yang

Merge develop tiny fix

上级 fdc68914
...@@ -12,11 +12,11 @@ ...@@ -12,11 +12,11 @@
See the License for the specific language governing permissions and See the License for the specific language governing permissions and
limitations under the License. */ limitations under the License. */
#include "paddle/fluid/framework/data_layout_transform.h"
#include "paddle/fluid/memory/malloc.h"
#include "paddle/fluid/operators/conv_op.h" #include "paddle/fluid/operators/conv_op.h"
#include "paddle/fluid/platform/mkldnn_helper.h" #include "paddle/fluid/platform/mkldnn_helper.h"
#include "paddle/fluid/framework/data_layout_transform.h"
namespace paddle { namespace paddle {
namespace operators { namespace operators {
...@@ -426,8 +426,9 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> { ...@@ -426,8 +426,9 @@ class ConvMKLDNNOpKernel : public paddle::framework::OpKernel<T> {
"same dimension sizes"); "same dimension sizes");
if (residual_param->format() != handler.GetDstFormat()) { if (residual_param->format() != handler.GetDstFormat()) {
auto output_data = auto output_data = output->mutable_data<T>(
output->mutable_data<T>(ctx.GetPlace(), handler.GetDstMemorySize()); ctx.GetPlace(), ::paddle::memory::Allocator::kDefault,
handler.GetDstMemorySize());
auto residual_data_tz = auto residual_data_tz =
paddle::framework::vectorize2int(residual_param->dims()); paddle::framework::vectorize2int(residual_param->dims());
auto residual_data_type = auto residual_data_type =
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册