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:
fa2213f
)
Revert "fix cuda half math function is undefined: hpow, htanh (#6225)" (#6249)
author
Tianqi Chen
<tqchen@users.noreply.github.com>
Wed, 12 Aug 2020 00:32:57 +0000
(17:32 -0700)
committer
GitHub
<noreply@github.com>
Wed, 12 Aug 2020 00:32:57 +0000
(17:32 -0700)
This reverts commit
ed04cdd35f1990959ec788be0131b1388fd11d31
.
src/target/source/literal/cuda_half_t.h
patch
|
blob
|
history
diff --git
a/src/target/source/literal/cuda_half_t.h
b/src/target/source/literal/cuda_half_t.h
index
422d2c0
..
baf4ba7
100644
(file)
--- a/
src/target/source/literal/cuda_half_t.h
+++ b/
src/target/source/literal/cuda_half_t.h
@@
-293,19
+293,6
@@
__pack_half2(const half x, const half y) {
unsigned v1 = *((unsigned short *)&y);
return (v1 << 16) | v0;
}
-
-static inline __device__ __host__ half hpow(half x, half y) {
- float tmp_x = __half2float(x);
- float tmp_y = __half2float(y);
- float result = powf(tmp_x, tmp_y);
- return __float2half(result);
-}
-
-static inline __device__ __host__ half htanh(half x) {
- float tmp_x = __half2float(x);
- float result = tanhf(tmp_x);
- return __float2half(result);
-}
)";
static constexpr const char* _cuda_warp_intrinsic_util = R"(