diff --git a/clang/lib/Sema/SemaExpr.cpp b/clang/lib/Sema/SemaExpr.cpp index 0cd86dc54b0ab..46e403b6db9a1 100644 --- a/clang/lib/Sema/SemaExpr.cpp +++ b/clang/lib/Sema/SemaExpr.cpp @@ -6357,6 +6357,14 @@ static FunctionDecl *rewriteBuiltinFunctionDecl(Sema *Sema, ASTContext &Context, Params.push_back(Parm); } OverloadDecl->setParams(Params); + // We cannot merge host/device attributes of redeclarations. They have to + // be consistent when created. + if (Sema->LangOpts.CUDA) { + if (FDecl->hasAttr()) + OverloadDecl->addAttr(CUDAHostAttr::CreateImplicit(Context)); + if (FDecl->hasAttr()) + OverloadDecl->addAttr(CUDADeviceAttr::CreateImplicit(Context)); + } Sema->mergeDeclAttributes(OverloadDecl, FDecl); return OverloadDecl; } diff --git a/clang/test/SemaCUDA/overloaded-builtin.cu b/clang/test/SemaCUDA/overloaded-builtin.cu new file mode 100644 index 0000000000000..c60c27e7f8627 --- /dev/null +++ b/clang/test/SemaCUDA/overloaded-builtin.cu @@ -0,0 +1,36 @@ +// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -aux-triple amdgcn-amd-amdhsa -fsyntax-only -verify=host -xhip %s +// RUN: %clang_cc1 -triple amdgcn-amd-amdhsa -fsyntax-only -fcuda-is-device -verify=dev -xhip %s + +// dev-no-diagnostics + +#include "Inputs/cuda.h" + +__global__ void kernel() { + __attribute__((address_space(0))) void *mem_ptr; + (void)__builtin_amdgcn_is_shared(mem_ptr); +} + +template +__global__ void template_kernel(T *p) { + __attribute__((address_space(0))) void *mem_ptr; + (void)__builtin_amdgcn_is_shared(mem_ptr); +} + +void hfun() { + __attribute__((address_space(0))) void *mem_ptr; + (void)__builtin_amdgcn_is_shared(mem_ptr); // host-error {{reference to __device__ function '__builtin_amdgcn_is_shared' in __host__ function}} +} + +template +void template_hfun(T *p) { + __attribute__((address_space(0))) void *mem_ptr; + (void)__builtin_amdgcn_is_shared(mem_ptr); // host-error {{reference to __device__ function '__builtin_amdgcn_is_shared' in __host__ function}} +} + + +int main() { + int *p; + kernel<<<1,1>>>(); + template_kernel<<<1,1>>>(p); + template_hfun(p); // host-note {{called by 'main'}} +}