From 60521a92c9fe7fd280cba2c51a9f29e8f17f70a6 Mon Sep 17 00:00:00 2001 From: Chenhu Wang Date: Mon, 4 Apr 2022 15:52:22 +0800 Subject: [PATCH] [CPU] Fixed NMS implementation on APL targets (#10649) --- src/plugins/intel_cpu/src/nodes/non_max_suppression.cpp | 2 +- src/plugins/intel_cpu/thirdparty/mkl-dnn | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/plugins/intel_cpu/src/nodes/non_max_suppression.cpp b/src/plugins/intel_cpu/src/nodes/non_max_suppression.cpp index 247ae52f83d..b4ca8b14e05 100644 --- a/src/plugins/intel_cpu/src/nodes/non_max_suppression.cpp +++ b/src/plugins/intel_cpu/src/nodes/non_max_suppression.cpp @@ -394,7 +394,7 @@ private: } else { // pure sse path, make sure don't spoil vmm_temp3, which may used in after soft-suppression uni_vmovups(vmm_temp4, vmm_temp3); - cmpps(vmm_temp4, vmm_iou_threshold, 0x07); // order compare, 0 for unorders + cmpps(vmm_temp4, vmm_iou_threshold, 0x07); // order compare, 0 for at least one is NaN uni_vmovups(vmm_temp2, vmm_temp3); cmpps(vmm_temp2, vmm_iou_threshold, 0x05); // _CMP_GE_US on sse, no direct _CMP_GE_OS supported. diff --git a/src/plugins/intel_cpu/thirdparty/mkl-dnn b/src/plugins/intel_cpu/thirdparty/mkl-dnn index 82ca2f931c1..8f988921d01 160000 --- a/src/plugins/intel_cpu/thirdparty/mkl-dnn +++ b/src/plugins/intel_cpu/thirdparty/mkl-dnn @@ -1 +1 @@ -Subproject commit 82ca2f931c1d588b67d154d873136d4af1ffb3a8 +Subproject commit 8f988921d019bdba8e0155a03fec9cedce75c54f