projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
97ab77e
)
Address issue #6415 using compiler-rt half-float function. (#6431)
author
Balint Cristian
<cristian.balint@gmail.com>
Thu, 10 Sep 2020 00:01:33 +0000
(
03:01
+0300)
committer
GitHub
<noreply@github.com>
Thu, 10 Sep 2020 00:01:33 +0000
(17:01 -0700)
src/relay/transforms/pattern_util.h
patch
|
blob
|
history
diff --git
a/src/relay/transforms/pattern_util.h
b/src/relay/transforms/pattern_util.h
index
f493720
..
39fbec5
100644
(file)
--- 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) {