[RISCV] [2/2] Add IR intrinsic for Zbm extension

[LevyHsu]

Summary

RV64 ONLY:

bmator
bmatxor
bmatflip

clang-tools-extra/clang-include-fixer/find-all-symbols/STLPostfixHeaderMap.cpp

static const HeaderMapCollector::RegexHeaderMap STLPostfixHeaderMap = {
       {"include/mmintrin.h$", "<mmintrin>"},
       {"include/mwaitxintrin.h$", "<x86intrin.h>"},
       {"include/pkuintrin.h$", "<immintrin.h>"},
       {"include/pmmintrin.h$", "<pmmintrin.h>"},
       {"include/popcntintrin.h$", "<popcntintrin.h>"},
       {"include/prfchwintrin.h$", "<x86intrin.h>"},
       {"include/rdseedintrin.h$", "<x86intrin.h>"},
       {"include/rtmintrin.h$", "<immintrin.h>"},
+      {"include/rvintrin.h$", "<rvintrin.h>"},
+      {"include/riscv_zbm_intrin.h$", "<riscv_zbm_intrin.h>"},
       {"include/shaintrin.h$", "<immintrin.h>"},
       {"include/smmintrin.h$", "<smmintrin.h>"},
       {"include/stdalign.h$", "<cstdalign>"},
       {"include/stdarg.h$", "<cstdarg>"},
       {"include/stdbool.h$", "<cstdbool>"},
       {"include/stddef.h$", "<cstddef>"},
       {"include/stdint.h$", "<cstdint>"},
       {"include/tbmintrin.h$", "<x86intrin.h>"},

clang-tools-extra/clangd/index/CanonicalIncludes.cpp

static const auto *SystemHeaderMap = new llvm::StringMap<llvm::StringRef>({
       {"include/mmintrin.h", "<mmintrin>"},
       {"include/mwaitxintrin.h", "<x86intrin.h>"},
       {"include/pkuintrin.h", "<immintrin.h>"},
       {"include/pmmintrin.h", "<pmmintrin.h>"},
       {"include/popcntintrin.h", "<popcntintrin.h>"},
       {"include/prfchwintrin.h", "<x86intrin.h>"},
       {"include/rdseedintrin.h", "<x86intrin.h>"},
       {"include/rtmintrin.h", "<immintrin.h>"},
+      {"include/rvintrin.h", "<rvintrin.h>"},
+      {"include/riscv_zbm_intrin.h", "<riscv_zbm_intrin.h>"},
       {"include/shaintrin.h", "<immintrin.h>"},
       {"include/smmintrin.h", "<smmintrin.h>"},
       {"include/stdalign.h", "<cstdalign>"},
       {"include/stdarg.h", "<cstdarg>"},
       {"include/stdbool.h", "<cstdbool>"},
       {"include/stddef.h", "<cstddef>"},
       {"include/stdint.h", "<cstdint>"},
       {"include/tbmintrin.h", "<x86intrin.h>"},

clang/lib/Headers/CMakeLists.txt

set(files
   pkuintrin.h
   pmmintrin.h
   pconfigintrin.h
   popcntintrin.h
   prfchwintrin.h
   ptwriteintrin.h
   rdseedintrin.h
   rtmintrin.h
+  rvintrin.h
+  riscv_zbm_intrin.h
   serializeintrin.h
   sgxintrin.h
   s390intrin.h
   shaintrin.h
   smmintrin.h
   stdalign.h
   stdarg.h
   stdatomic.h

clang/lib/Headers/riscv_zbm_intrin.h

+/*===------------------------ riscv_zbm_intrin.h ---------------------------===
+ *
+ * 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
+ *
+ *===-----------------------------------------------------------------------===
+ */
+
+#ifndef __RVINTRIN_H
+#error "Never use <riscv_zbm_intrin.h> directly; include <rvintrin.h> instead."
+#endif
+
+#ifndef __RISCV_ZBM_INTRIN_H
+#define __RISCV_ZBM_INTRIN_H
+
+#include <stdint.h>
+
+#if defined(__cplusplus)
+extern "C" {
+#endif
+
+// Zbe RV64 ONLY
+#if __riscv_xlen == 64
+static __inline__ int64_t __DEFAULT_FN_ATTRS _rv_bmator(int64_t rs1,
+                                                        int64_t rs2) {
+  return __builtin_riscv_bmator(rs1, rs2);
+}
+
+static __inline__ int64_t __DEFAULT_FN_ATTRS _rv__bmatxor(int64_t rs1,
+                                                          int64_t rs2) {
+  return __builtin_riscv_bmatxor(rs1, rs2);
+}
+
+static __inline__ int64_t __DEFAULT_FN_ATTRS _rv__bmatflip(int64_t rs1) {
+  return __builtin_riscv_bmatflip(rs1);
+}
+#endif // if __riscv_xlen == 64
+
+#if defined(__cplusplus)
+}
+#endif // if defined(__cplusplus)
+
+#endif // __RISCV_ZBM_INTRIN_H

clang/lib/Headers/rvintrin.h

+/*===-------------------------- rvintrin.h ---------------------------------===
+ *
+ * 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
+ *
+ *===-----------------------------------------------------------------------===
+ */
+
+#ifndef __RVINTRIN_H
+#define __RVINTRIN_H
+
+#define int_xlen_t long
+#define uint_xlen_t unsigned int_xlen_t
+
+#define __DEFAULT_FN_ATTRS                                                     \
+  __attribute__((__always_inline__, __artificial__, __nodebug__))
+
+#if defined(__riscv_zbm)
+#include "riscv_zbm_intrin.h"
+#endif
+
+#undef __DEFAULT_FN_ATTRS
+#undef uint_xlen_t
+#undef int_xlen_t
+#endif // __RVINTRIN_H
 No newline at end of file

clang/test/CodeGen/RISCV/rvb-intrinsics/rvintrin.c

+// RUN: %clang_cc1 -triple riscv32 -fsyntax-only \
+// RUN:   -target-feature +experimental-zbm %s
+// RUN: %clang_cc1 -triple riscv64 -fsyntax-only \
+// RUN:   -target-feature +experimental-zbm %s
+
+#include <rvintrin.h>
 No newline at end of file