use hip rsqrt function

ee1cc901 · Khalique · 86094d23 · ee1cc901
Commit ee1cc901 authored Jul 16, 2019 by Khalique
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/targets/gpu/device/rsqrt.cpp src/targets/gpu/device/rsqrt.cpp +1 -1

No files found.
--- a/src/targets/gpu/device/rsqrt.cpp
+++ b/src/targets/gpu/device/rsqrt.cpp
@@ -9,7 +9,7 @@ namespace device {

 void rsqrt(hipStream_t stream, const argument& result, const argument& arg)
 {
-    nary(stream, result, arg)([](auto x) { return 1 / ::sqrt(to_hip_type(x)); });
+    nary(stream, result, arg)([](auto x) __device__ { return ::rsqrt(to_hip_type(x)); });
 }

 } // namespace device