saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-25 02:50:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove ROCM workaround for half-to-double cast.

Browse source
This commit is contained in:
Jesse Benson 2021-04-06 14:17:21 -07:00 committed by Jesse Benson
parent 25e261f196
commit 2ec452cdad
1 changed files with 1 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
onnxruntime/core/providers/rocm/cu_inc/common.cuh
View file

@ -144,18 +144,10 @@ __device__ __inline__ half2 _Tanh(half2 a) {
return __float22half2_rn(tmp);
}
// TODO: temporary workaround for casting half-to-double, until ROCM/hipcc adds support.
namespace {
template <typename T>
__device__ __inline__ double __cast_to_double(T x) { return static_cast<double>(x); }
template <>
__device__ __inline__ double __cast_to_double(half x) { return static_cast<double>(static_cast<float>(x)); }
} // namespace
// Capture permutations of int32/64/float/double
template <typename T, typename T1>
__device__ __inline__ T _Pow(T a, T1 b) {
return static_cast<T>(pow(__cast_to_double(a), __cast_to_double(b)));
return static_cast<T>(pow(static_cast<double>(a), static_cast<double>(b)));
}
template <>