[libspirv] Add __spirv_BitFieldUExtract/SExtract/Insert implementation (#20450)

wenju-he · web-flow · commit a3f4959c0c40 · 2025-10-31T20:30:51.000-07:00
diff --git a/libclc/libspirv/lib/generic/SOURCES b/libclc/libspirv/lib/generic/SOURCES
@@ -39,6 +39,9 @@ geometric/scale.cl
 integer/abs.cl
 integer/abs_diff.cl
 integer/add_sat.cl
+integer/bitfield_extract_signed.cl
+integer/bitfield_extract_unsigned.cl
+integer/bitfield_insert.cl
 integer/bit_reverse.cl
 integer/clamp.cl
 integer/clz.cl
diff --git a/libclc/libspirv/lib/generic/integer/bitfield_extract_signed.cl b/libclc/libspirv/lib/generic/integer/bitfield_extract_signed.cl
@@ -0,0 +1,13 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include <clc/integer/clc_bitfield_extract_signed.h>
+#include <libspirv/spirv.h>
+
+#define __CLC_BODY <bitfield_extract_signed.inc>
+#include <clc/integer/gentype.inc>
diff --git a/libclc/libspirv/lib/generic/integer/bitfield_extract_signed.inc b/libclc/libspirv/lib/generic/integer/bitfield_extract_signed.inc
@@ -0,0 +1,16 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifdef __CLC_GEN_S
+
+_CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE
+__spirv_BitFieldSExtract(__CLC_GENTYPE base, uint offset, uint count) {
+  return __clc_bitfield_extract_signed(base, offset, count);
+}
+
+#endif
diff --git a/libclc/libspirv/lib/generic/integer/bitfield_extract_unsigned.cl b/libclc/libspirv/lib/generic/integer/bitfield_extract_unsigned.cl
@@ -0,0 +1,13 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include <clc/integer/clc_bitfield_extract_unsigned.h>
+#include <libspirv/spirv.h>
+
+#define __CLC_BODY <bitfield_extract_unsigned.inc>
+#include <clc/integer/gentype.inc>
diff --git a/libclc/libspirv/lib/generic/integer/bitfield_extract_unsigned.inc b/libclc/libspirv/lib/generic/integer/bitfield_extract_unsigned.inc
@@ -0,0 +1,16 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#ifdef __CLC_GEN_U
+
+_CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE
+__spirv_BitFieldUExtract(__CLC_GENTYPE base, uint offset, uint count) {
+  return __clc_bitfield_extract_unsigned(base, offset, count);
+}
+
+#endif
diff --git a/libclc/libspirv/lib/generic/integer/bitfield_insert.cl b/libclc/libspirv/lib/generic/integer/bitfield_insert.cl
@@ -0,0 +1,13 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include <clc/integer/clc_bitfield_insert.h>
+#include <libspirv/spirv.h>
+
+#define __CLC_BODY <bitfield_insert.inc>
+#include <clc/integer/gentype.inc>
diff --git a/libclc/libspirv/lib/generic/integer/bitfield_insert.inc b/libclc/libspirv/lib/generic/integer/bitfield_insert.inc
@@ -0,0 +1,12 @@
+//===----------------------------------------------------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+_CLC_OVERLOAD _CLC_DEF __CLC_GENTYPE __spirv_BitFieldInsert(
+    __CLC_GENTYPE base, __CLC_GENTYPE insert, uint offset, uint count) {
+  return __clc_bitfield_insert(base, insert, offset, count);
+}
