diff options
| author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2019-08-16 18:03:35 +0100 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2019-08-19 12:37:27 +0000 |
| commit | 81d7e78bb3c7291018be30efc88467979482cc65 (patch) | |
| tree | 20b8025c39136e39d6288327badac85bac3d3245 /src | |
| parent | c755f78e366443ffbe168bf5096c96b8121304b9 (diff) | |
| download | ComputeLibrary-81d7e78bb3c7291018be30efc88467979482cc65.tar.gz | |
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>
Diffstat (limited to 'src')
| -rw-r--r-- | src/core/NEON/kernels/NEReductionOperationKernel.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
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 { |