[clang] [llvm] [clang] Simplify device kernel attributes (PR #137882)
Nick Sarnie via llvm-commits
llvm-commits at lists.llvm.org
Thu May 8 13:15:33 PDT 2025
================
@@ -1538,11 +1533,44 @@ def CUDAShared : InheritableAttr {
}
def : MutualExclusions<[CUDAConstant, CUDAShared, HIPManaged]>;
-def SYCLKernel : InheritableAttr {
- let Spellings = [Clang<"sycl_kernel">];
- let Subjects = SubjectList<[FunctionTmpl]>;
- let LangOpts = [SYCLDevice];
+def DeviceKernel : DeclOrTypeAttr {
+ let Spellings = [Clang<"device_kernel">, Clang<"sycl_kernel">,
+ Clang<"nvptx_kernel">, Clang<"amdgpu_kernel">,
+ CustomKeyword<"__kernel">, CustomKeyword<"kernel">];
+ let LangOpts = [];
let Documentation = [SYCLKernelDocs];
+ let AdditionalMembers =
+ [{
+ inline bool isAMDGPUSpelling() const {
+ return isAMDGPUSpelling(*this);
+ }
+ template<typename T>
+ static inline bool isAMDGPUSpelling(const T& Attr) {
+ return Attr.getAttrName()->getName() == "amdgpu_kernel";
+ }
+ inline bool isNVPTXSpelling() const {
+ return isNVPTXSpelling(*this);
+ }
+ template<typename T>
+ static inline bool isNVPTXSpelling(const T& Attr) {
+ return Attr.getAttrName()->getName() == "nvptx_kernel";
+ }
+ inline bool isOpenCLSpelling() const {
+ return isOpenCLSpelling(*this);
+ }
+ template<typename T>
+ static inline bool isOpenCLSpelling(const T& Attr) {
+ return Attr.getAttrName()->getName() == "kernel" ||
----------------
sarnex wrote:
I ended up doing it this way using the enums.
https://github.com/llvm/llvm-project/pull/137882
More information about the llvm-commits
mailing list