diff --git a/clang-tools-extra/clang-include-fixer/find-all-symbols/STLPostfixHeaderMap.cpp b/clang-tools-extra/clang-include-fixer/find-all-symbols/STLPostfixHeaderMap.cpp --- a/clang-tools-extra/clang-include-fixer/find-all-symbols/STLPostfixHeaderMap.cpp +++ b/clang-tools-extra/clang-include-fixer/find-all-symbols/STLPostfixHeaderMap.cpp @@ -57,6 +57,8 @@ {"include/prfchwintrin.h$", ""}, {"include/rdseedintrin.h$", ""}, {"include/rtmintrin.h$", ""}, + {"include/rvintrin.h$", ""}, + {"include/riscv_zbb_intrin.h$", ""}, {"include/shaintrin.h$", ""}, {"include/smmintrin.h$", ""}, {"include/stdalign.h$", ""}, diff --git a/clang-tools-extra/clangd/index/CanonicalIncludes.cpp b/clang-tools-extra/clangd/index/CanonicalIncludes.cpp --- a/clang-tools-extra/clangd/index/CanonicalIncludes.cpp +++ b/clang-tools-extra/clangd/index/CanonicalIncludes.cpp @@ -152,6 +152,8 @@ {"include/prfchwintrin.h", ""}, {"include/rdseedintrin.h", ""}, {"include/rtmintrin.h", ""}, + {"include/rvintrin.h", ""}, + {"include/riscv_zbb_intrin.h", ""}, {"include/shaintrin.h", ""}, {"include/smmintrin.h", ""}, {"include/stdalign.h", ""}, diff --git a/clang/lib/Headers/CMakeLists.txt b/clang/lib/Headers/CMakeLists.txt --- a/clang/lib/Headers/CMakeLists.txt +++ b/clang/lib/Headers/CMakeLists.txt @@ -97,6 +97,8 @@ ptwriteintrin.h rdseedintrin.h rtmintrin.h + rvintrin.h + riscv_zbb_intrin.h serializeintrin.h sgxintrin.h s390intrin.h diff --git a/clang/lib/Headers/riscv_zbb_intrin.h b/clang/lib/Headers/riscv_zbb_intrin.h new file mode 100644 --- /dev/null +++ b/clang/lib/Headers/riscv_zbb_intrin.h @@ -0,0 +1,26 @@ +/* ===-------- riscv_zbb_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 __RISCV_ZBB_INTRIN_H +#define __RISCV_ZBB_INTRIN_H + +#if defined(__cplusplus) +extern "C" { +#endif + +// Zbb +static __inline__ int_xlen_t __DEFAULT_FN_ATTRS _rv_orc_b(int_xlen_t rs1) { + return __builtin_riscv_orc_b(rs1); +} + +#if defined(__cplusplus) +} +#endif // if defined(__cplusplus) + +#endif // __RISCV_ZBB_INTRIN_H diff --git a/clang/lib/Headers/rvintrin.h b/clang/lib/Headers/rvintrin.h new file mode 100644 --- /dev/null +++ b/clang/lib/Headers/rvintrin.h @@ -0,0 +1,28 @@ +/* ===-------- 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 + +// Long is 32 bit on riscv32 and 64 bit on riscv64 according to calling +// convention. +#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_zbb) +#include "riscv_zbb_intrin.h" +#endif + +#undef __DEFAULT_FN_ATTRS +#undef uint_xlen_t +#undef int_xlen_t +#endif // __RVINTRIN_H diff --git a/clang/lib/Sema/SemaChecking.cpp b/clang/lib/Sema/SemaChecking.cpp --- a/clang/lib/Sema/SemaChecking.cpp +++ b/clang/lib/Sema/SemaChecking.cpp @@ -3395,13 +3395,28 @@ CallExpr *TheCall) { // CodeGenFunction can also detect this, but this gives a better error // message. + bool miss_feature_error = false; + SmallVector ReqFeatures; StringRef Features = Context.BuiltinInfo.getRequiredFeatures(BuiltinID); - if (Features.find("experimental-v") != StringRef::npos && - !TI.hasFeature("experimental-v")) - return Diag(TheCall->getBeginLoc(), diag::err_riscvv_builtin_requires_v) - << TheCall->getSourceRange(); - - return false; + StringRef(Features).split(ReqFeatures, ','); + + // Check if each required feature is included in TargetInfo + for (size_t i = 0; i < ReqFeatures.size(); i++) { + if (!TI.hasFeature(ReqFeatures[i])) { + + // Make message like "experimental-zbr" to "Zbr" + ReqFeatures[i].consume_front("experimental-"); + std::string FeatureStr = ReqFeatures[i].str(); + FeatureStr[0] = std::toupper(FeatureStr[0]); + + // Error message + miss_feature_error = true; + Diag(TheCall->getBeginLoc(), diag::err_riscv_builtin_requires_extension) + << TheCall->getSourceRange() << StringRef(FeatureStr); + } + } + + return miss_feature_error; } bool Sema::CheckSystemZBuiltinFunctionCall(unsigned BuiltinID, diff --git a/clang/test/Headers/rvintrin.c b/clang/test/Headers/rvintrin.c new file mode 100644 --- /dev/null +++ b/clang/test/Headers/rvintrin.c @@ -0,0 +1,6 @@ +// RUN: %clang_cc1 -triple riscv32 -fsyntax-only \ +// RUN: -target-feature +experimental-zbb %s +// RUN: %clang_cc1 -triple riscv64 -fsyntax-only \ +// RUN: -target-feature +experimental-zbb %s + +#include