diff --git a/paddle/fluid/operators/elementwise/mkldnn/elementwise_add_mkldnn_op.cc b/paddle/fluid/operators/elementwise/mkldnn/elementwise_add_mkldnn_op.cc index 2779f6dd9ce2299de094ec64f2555a16c972ef35..88cda1cd66868933de99b3b864ea98627df1e304 100644 --- a/paddle/fluid/operators/elementwise/mkldnn/elementwise_add_mkldnn_op.cc +++ b/paddle/fluid/operators/elementwise/mkldnn/elementwise_add_mkldnn_op.cc @@ -70,8 +70,7 @@ class EltwiseAddMKLDNNKernel : public framework::OpKernel { auto x_memory_pd = memory::primitive_desc( {{src_x_tz}, memory::data_type::f32, format}, mkldnn_engine); auto size = x_memory_pd.get_size(); - _x.mutable_data(ctx.GetPlace(), paddle::memory::Allocator::kDefault, - size); + _x.mutable_data(ctx.GetPlace(), size); auto user_x_memory = memory(user_x_memory_pd, paddle::platform::to_void_cast(x_data)); auto x_memory = memory(x_memory_pd,