You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2020/09/10 00:01:55 UTC

[incubator-tvm] branch master updated: Address issue #6415 using compiler-rt half-float function. (#6431)

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new 29cf2e0  Address issue #6415 using compiler-rt half-float function. (#6431)
29cf2e0 is described below

commit 29cf2e08efdfe0125442cdfd76868f8eb59b72d3
Author: Balint Cristian <cr...@gmail.com>
AuthorDate: Thu Sep 10 03:01:33 2020 +0300

    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<uint64_t*>(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__<uint16_t, uint16_t, 10, float, uint32_t, 23>(
+          reinterpret_cast<uint16_t*>(array->data)[i]);
     }
-#endif
     if (array->dtype.bits == 32) {
       return reinterpret_cast<float*>(array->data)[i];
     } else if (array->dtype.bits == 64) {