diff --git a/clang/lib/Sema/SemaType.cpp b/clang/lib/Sema/SemaType.cpp --- a/clang/lib/Sema/SemaType.cpp +++ b/clang/lib/Sema/SemaType.cpp @@ -3940,6 +3940,20 @@ break; } } + } else if (S.getLangOpts().CUDA) { + // If we're compiling CUDA/HIP code and targeting SPIR-V we need to make + // sure the kernels will be marked with the right calling convention so that + // they will be visible by the APIs that ingest SPIR-V. + llvm::Triple Triple = S.Context.getTargetInfo().getTriple(); + if (Triple.getArch() == llvm::Triple::spirv32 || + Triple.getArch() == llvm::Triple::spirv64) { + for (const ParsedAttr &AL : D.getDeclSpec().getAttributes()) { + if (AL.getKind() == ParsedAttr::AT_CUDAGlobal) { + CC = CC_OpenCLKernel; + break; + } + } + } } return CC; diff --git a/clang/test/CodeGenCUDASPIRV/kernel-cc.cu b/clang/test/CodeGenCUDASPIRV/kernel-cc.cu new file mode 100644 --- /dev/null +++ b/clang/test/CodeGenCUDASPIRV/kernel-cc.cu @@ -0,0 +1,9 @@ +// RUN: %clang_cc1 -fcuda-is-device -triple spirv32 -o - -emit-llvm -x cuda %s | FileCheck %s +// RUN: %clang_cc1 -fcuda-is-device -triple spirv64 -o - -emit-llvm -x cuda %s | FileCheck %s + +// Verifies that building CUDA targeting SPIR-V {32,64} generates LLVM IR with +// spir_kernel attributes for kernel functions. + +// CHECK: define spir_kernel void @_Z6kernelv() + +__attribute__((global)) void kernel() { return; }