diff --git a/clang/lib/CodeGen/CGCUDANV.cpp b/clang/lib/CodeGen/CGCUDANV.cpp --- a/clang/lib/CodeGen/CGCUDANV.cpp +++ b/clang/lib/CodeGen/CGCUDANV.cpp @@ -1195,8 +1195,16 @@ llvm::GlobalValue *CGNVCUDARuntime::getKernelHandle(llvm::Function *F, GlobalDecl GD) { auto Loc = KernelHandles.find(F->getName()); - if (Loc != KernelHandles.end()) - return Loc->second; + if (Loc != KernelHandles.end()) { + if (Loc->second == F || CGM.getLangOpts().HIP) + return Loc->second; + // non-HIP compilation may end up with a different F and need to have + // handles and stubs updated. + KernelStubs.erase(Loc->second); + KernelStubs[F] = F; + KernelHandles[F->getName()] = F; + return F; + } if (!CGM.getLangOpts().HIP) { KernelHandles[F->getName()] = F; diff --git a/clang/test/CodeGenCUDA/bug-kerner-registration-reuse.cu b/clang/test/CodeGenCUDA/bug-kerner-registration-reuse.cu new file mode 100644 --- /dev/null +++ b/clang/test/CodeGenCUDA/bug-kerner-registration-reuse.cu @@ -0,0 +1,24 @@ +// RUN: echo -n "GPU binary would be here." > %t +// RUN: %clang_cc1 -triple x86_64-linux-gnu -emit-llvm %s \ +// RUN: -target-sdk-version=11.0 -fcuda-include-gpubinary %t -o - \ +// RUN: | FileCheck %s + +#include "Inputs/cuda.h" + +template +struct S { T t; }; + +template +static __global__ void Kernel(S) {} + +// For some reason it takes three or more instantiations of Kernel to trigger a +// crash. +auto x = &Kernel; +auto y = &Kernel; +auto z = &Kernel; + +// CHECK-LABEL: @__cuda_register_globals( +// CHECK: call i32 @__cudaRegisterFunction(ptr %0, ptr @_ZL21__device_stub__KernelIdEv1SIT_E +// CHECK: call i32 @__cudaRegisterFunction(ptr %0, ptr @_ZL21__device_stub__KernelIfEv1SIT_E +// CHECK: call i32 @__cudaRegisterFunction(ptr %0, ptr @_ZL21__device_stub__KernelIiEv1SIT_E +// CHECK: ret void