llvm/clang/test/CodeGenCUDA/bug-kerner-registration-reuse.cu

// 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 --check-prefixes CUDA
// RUN: %clang_cc1 -triple x86_64-linux-gnu -emit-llvm %s -x hip \
// RUN:     -fcuda-include-gpubinary %t -o - \
// RUN: | FileCheck %s --check-prefixes HIP

#include "Inputs/cuda.h"

template <typename T>
struct S { T t; };

template <typename T>
    static __global__ void Kernel(S<T>) {}

// For some reason it takes three or more instantiations of Kernel to trigger a
// crash during CUDA compilation.
auto x = &Kernel<double>;
auto y = &Kernel<float>;
auto z = &Kernel<int>;

// This triggers HIP-specific code path.
void func (){
  Kernel<short><<<1,1>>>({1});
}

// CUDA-LABEL: @__cuda_register_globals(
// CUDA:  call i32 @__cudaRegisterFunction(ptr %0, ptr @_ZL21__device_stub__KernelIdEv1SIT_E
// CUDA:  call i32 @__cudaRegisterFunction(ptr %0, ptr @_ZL21__device_stub__KernelIfEv1SIT_E
// CUDA:  call i32 @__cudaRegisterFunction(ptr %0, ptr @_ZL21__device_stub__KernelIiEv1SIT_E
// CUDA:  ret void

// HIP-LABEL: @__hip_register_globals(
// HIP:   call i32 @__hipRegisterFunction(ptr %0, ptr @_ZL6KernelIdEv1SIT_E
// HIP:   call i32 @__hipRegisterFunction(ptr %0, ptr @_ZL6KernelIfEv1SIT_E
// HIP:   call i32 @__hipRegisterFunction(ptr %0, ptr @_ZL6KernelIiEv1SIT_E
// HIP:   ret void