[clang] [llvm] [clang] Add spir_kernel attribute (PR #137882)
Aaron Ballman via llvm-commits
llvm-commits at lists.llvm.org
Thu May 8 05:17:07 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);
+ }
----------------
AaronBallman wrote:
There's an easier way to do this, I think. We have:
```
let Accessors = [Accessor<"isAMDGPU", [Clang<"amdgpu_kernel">]>,
Accessor<"isNVPTX", [Clang<"nvptx_kernel">]>];
```
which generates an accessor method for you which returns true/false based on the spelling.
https://github.com/llvm/llvm-project/pull/137882
More information about the llvm-commits
mailing list