"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "27b8491640aac89a08624f3f70a270ee88542984"
Unverified Commit 67497a04 authored by Jatin Chaudhary's avatar Jatin Chaudhary Committed by GitHub
Browse files

Make sure we call __hneg with half to remove ambigios error (#1736)

parent 94ae7113
...@@ -611,7 +611,7 @@ inline __device__ int8_t neg<int8_t>(int8_t x) ...@@ -611,7 +611,7 @@ inline __device__ int8_t neg<int8_t>(int8_t x)
template <> template <>
inline __device__ half_t neg<half_t>(half_t x) inline __device__ half_t neg<half_t>(half_t x)
{ {
return __hneg(x); return __hneg(static_cast<__half>(x));
}; };
template <typename T> template <typename T>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment