llvm-project/clang/test/SemaCUDA/overloaded-builtin.cu
Yaxun (Sam) Liu c16297cd3f
[CUDA][HIP] Fix host/device attribute of builtin (#138162)
When a builtin function is passed a pointer with a different
address space, clang creates an overloaded
builtin function but does not copy the host/device attribute. This
causes
error when the builtin is called by device functions
since CUDA/HIP relies on the host/device attribute to treat
a builtin function as callable on both host and device
sides.

Fixed by copying the host/device attribute of the original
builtin function to the created overloaded builtin function.
2025-05-07 22:03:33 -04:00

37 lines
1.2 KiB
Plaintext

// 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<typename T>
__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<typename T>
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'}}
}