From 115851c8f51ede4ac4a42737ebd80d6f41fe8c33 Mon Sep 17 00:00:00 2001
From: Michele Di Giorgio <michele.digiorgio@arm.com>
Date: Fri, 16 Aug 2019 18:03:35 +0100
Subject: COMPMID-2595: (Nightly) Mismatches in NEArgMinMax for FP16

Results for FP16 were correctly computed but not stored to the output.

Change-Id: I8a85e325ee42600e0450307562c344b48eb1c679
Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1757
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
---
 src/core/NEON/kernels/NEReductionOperationKernel.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/core/NEON/kernels/NEReductionOperationKernel.cpp b/src/core/NEON/kernels/NEReductionOperationKernel.cpp
index 67ccc5d736..ffa4fa3565 100644
--- a/src/core/NEON/kernels/NEReductionOperationKernel.cpp
+++ b/src/core/NEON/kernels/NEReductionOperationKernel.cpp
@@ -809,6 +809,12 @@ struct RedOpYZW
             if(op == ReductionOperation::ARG_IDX_MIN || op == ReductionOperation::ARG_IDX_MAX)
             {
                 wrapper::vstore(reinterpret_cast<uint32_t *>(output.ptr()), vec_res_idx.val[0]);
+#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
+                if(std::is_same<T, float16_t>::value)
+                {
+                    wrapper::vstore(reinterpret_cast<uint32_t *>(output.ptr()) + 4, vec_res_idx.val[1]);
+                }
+#endif // __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
             }
             else
             {
-- 
cgit v1.2.1