From 29cf2e08efdfe0125442cdfd76868f8eb59b72d3 Mon Sep 17 00:00:00 2001 From: Balint Cristian Date: Thu, 10 Sep 2020 03:01:33 +0300 Subject: [PATCH] Address issue #6415 using compiler-rt half-float function. (#6431) --- src/relay/transforms/pattern_util.h | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/relay/transforms/pattern_util.h b/src/relay/transforms/pattern_util.h index f493720..39fbec5 100644 --- a/src/relay/transforms/pattern_util.h +++ b/src/relay/transforms/pattern_util.h @@ -396,11 +396,10 @@ static inline long double ToScalar(const runtime::NDArray& array, size_t i = 0) return reinterpret_cast(array->data)[i]; } } else if (array->dtype.code == kDLFloat) { -#if (__ARM_FP16_FORMAT_IEEE == 1) if (array->dtype.bits == 16) { - return reinterpret_cast<__fp16*>(array->data)[i]; + return __extendXfYf2__( + reinterpret_cast(array->data)[i]); } -#endif if (array->dtype.bits == 32) { return reinterpret_cast(array->data)[i]; } else if (array->dtype.bits == 64) { -- 2.7.4