libclc: Move rsqrt implementation to a .cl file
Reviewer: awatry
Differential Revision: https://reviews.llvm.org/D74013
GitOrigin-RevId: 4b23a2e8e971876d075d3ae322754dbc0495413d
diff --git a/generic/include/clc/math/rsqrt.h b/generic/include/clc/math/rsqrt.h
index 9d49ee6..41b9fd7 100644
--- a/generic/include/clc/math/rsqrt.h
+++ b/generic/include/clc/math/rsqrt.h
@@ -1 +1,7 @@
-#define rsqrt(x) (1.f/sqrt(x))
+#define __CLC_BODY <clc/math/unary_decl.inc>
+#define __CLC_FUNCTION rsqrt
+
+#include <clc/math/gentype.inc>
+
+#undef __CLC_BODY
+#undef __CLC_FUNCTION
diff --git a/generic/lib/SOURCES b/generic/lib/SOURCES
index df7f68f..ee2736b 100644
--- a/generic/lib/SOURCES
+++ b/generic/lib/SOURCES
@@ -176,6 +176,7 @@
math/clc_rootn.cl
math/rootn.cl
math/round.cl
+math/rsqrt.cl
math/sin.cl
math/sincos.cl
math/sincos_helpers.cl
diff --git a/generic/lib/math/rsqrt.cl b/generic/lib/math/rsqrt.cl
new file mode 100644
index 0000000..131ffc1
--- /dev/null
+++ b/generic/lib/math/rsqrt.cl
@@ -0,0 +1,23 @@
+#include <clc/clc.h>
+
+#include "../clcmacro.h"
+
+_CLC_OVERLOAD _CLC_DEF float rsqrt(float x)
+{
+ return 1.0f / sqrt(x);
+}
+
+_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, float, rsqrt, float);
+
+#ifdef cl_khr_fp64
+
+#pragma OPENCL EXTENSION cl_khr_fp64 : enable
+
+_CLC_OVERLOAD _CLC_DEF double rsqrt(double x)
+{
+ return 1.0 / sqrt(x);
+}
+
+_CLC_UNARY_VECTORIZE(_CLC_OVERLOAD _CLC_DEF, double, rsqrt, double);
+
+#endif
