From 3f49817a453ce834c17920302d8432350e2183b3 Mon Sep 17 00:00:00 2001 From: yeliang2258 <30516196+yeliang2258@users.noreply.github.com> Date: Wed, 10 Aug 2022 19:00:54 +0800 Subject: [PATCH] fix mkldnn interpolate ops (#45008) --- paddle/fluid/operators/mkldnn/interpolate_mkldnn_op.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/mkldnn/interpolate_mkldnn_op.cc b/paddle/fluid/operators/mkldnn/interpolate_mkldnn_op.cc index c3710342d80..64d7bca4d06 100644 --- a/paddle/fluid/operators/mkldnn/interpolate_mkldnn_op.cc +++ b/paddle/fluid/operators/mkldnn/interpolate_mkldnn_op.cc @@ -105,7 +105,8 @@ class InterpolateMKLDNNKernel : public framework::OpKernel { } } } - if (scale[0] > 0.0f && scale[1] > 0.0f && scale[2] > 0.0f) { + if (scale.size() == 3 && scale[0] > 0.0f && scale[1] > 0.0f && + scale[2] > 0.0f) { int j = 0; std::vector in_dhw_vec = phi::vectorize(in_dhw_dims); std::transform( -- GitLab