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) {