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 <[email protected]>
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) {
