[libclc] Move nan to the CLC library (#128521)

llvm · Feb 24, 2025 · 2dfb29a · 2dfb29a
1 parent a88167a
commit 2dfb29a
Show file tree

Hide file tree

Showing 7 changed files with 37 additions and 16 deletions.
diff --git a/libclc/clc/include/clc/math/clc_nan.h b/libclc/clc/include/clc/math/clc_nan.h
@@ -0,0 +1,12 @@
+#ifndef __CLC_MATH_CLC_NAN_H__
+#define __CLC_MATH_CLC_NAN_H__
+
+#define __CLC_FUNCTION __clc_nan
+#define __CLC_BODY <clc/math/clc_nan.inc>
+
+#include <clc/math/gentype.inc>
+
+#undef __CLC_BODY
+#undef __CLC_FUNCTION
+
+#endif // __CLC_MATH_CLC_NAN_H__
diff --git a/libclc/clc/include/clc/math/clc_nan.inc b/libclc/clc/include/clc/math/clc_nan.inc
@@ -0,0 +1 @@
+_CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE __clc_nan(__CLC_U_GENTYPE code);
diff --git a/libclc/clc/lib/generic/SOURCES b/libclc/clc/lib/generic/SOURCES
@@ -25,6 +25,7 @@ math/clc_floor.cl
 math/clc_frexp.cl
 math/clc_mad.cl
 math/clc_modf.cl
+math/clc_nan.cl
 math/clc_nextafter.cl
 math/clc_rint.cl
 math/clc_sw_fma.cl

diff --git a/libclc/clc/lib/generic/math/clc_nan.cl b/libclc/clc/lib/generic/math/clc_nan.cl
@@ -0,0 +1,5 @@
+#include <clc/internal/clc.h>
+#include <clc/utils.h>
+
+#define __CLC_BODY <clc_nan.inc>
+#include <clc/math/gentype.inc>
diff --git a/libclc/clc/lib/generic/math/clc_nan.inc b/libclc/clc/lib/generic/math/clc_nan.inc
@@ -0,0 +1,15 @@
+#if __CLC_FPSIZE == 64
+#define NAN_MASK 0x7ff0000000000000ul
+#elif __CLC_FPSIZE == 32
+#define NAN_MASK 0x7fc00000
+#elif __CLC_FPSIZE == 16
+#define NAN_MASK 0x7e00
+#endif
+
+_CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE __clc_nan(__CLC_U_GENTYPE code) {
+  const __CLC_U_GENTYPE mask = NAN_MASK;
+  const __CLC_U_GENTYPE res = code | mask;
+  return __CLC_AS_GENTYPE(res);
+}
+
+#undef NAN_MASK
diff --git a/libclc/generic/lib/math/nan.cl b/libclc/generic/lib/math/nan.cl
@@ -1,6 +1,5 @@
-#include <clc/clc.h>
-#include <clc/utils.h>
+#include <clc/clcmacro.h>
+#include <clc/math/clc_nan.h>
 
-#define __OPENCL_AS_GENTYPE __CLC_XCONCAT(as_, __CLC_GENTYPE)
 #define __CLC_BODY <nan.inc>
 #include <clc/math/gentype.inc>
diff --git a/libclc/generic/lib/math/nan.inc b/libclc/generic/lib/math/nan.inc
@@ -1,15 +1,3 @@
-#if __CLC_FPSIZE == 64
-#define NAN_MASK 0x7ff0000000000000ul
-#elif __CLC_FPSIZE == 32
-#define NAN_MASK 0x7fc00000
-#elif __CLC_FPSIZE == 16
-#define NAN_MASK 0x7e00
-#endif
-
 _CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE nan(__CLC_U_GENTYPE code) {
-  const __CLC_U_GENTYPE mask = NAN_MASK;
-  const __CLC_U_GENTYPE res = code | mask;
-  return __OPENCL_AS_GENTYPE(res);
+  return __clc_nan(code);
 }
-
-#undef NAN_MASK
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		_CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE __clc_nan(__CLC_U_GENTYPE code);