cc9477166a
Add builtin function __builtin_get_device_side_mangled_name to get device side manged name for functions and global variables, which can be used to get symbol address of kernels or variables by mangled name in dynamically loaded bundled code objects at run time. Reviewed by: Artem Belevich Differential Revision: https://reviews.llvm.org/D99301
29 lines
1.0 KiB
Plaintext
29 lines
1.0 KiB
Plaintext
// RUN: %clang_cc1 -triple x86_64-unknown-gnu-linux -aux-triple amdgcn-amd-amdhsa \
|
|
// RUN: -emit-llvm -o - -x hip %s | FileCheck -check-prefixes=CHECK,LNX %s
|
|
// RUN: %clang_cc1 -triple x86_64-unknown-windows-msvc -aux-triple amdgcn-amd-amdhsa \
|
|
// RUN: -emit-llvm -o - -x hip %s | FileCheck -check-prefixes=CHECK,MSVC %s
|
|
|
|
#include "Inputs/cuda.h"
|
|
|
|
namespace X {
|
|
__global__ void kern1(int *x);
|
|
__device__ int var1;
|
|
}
|
|
|
|
// CHECK: @[[STR1:.*]] = {{.*}} c"_ZN1X5kern1EPi\00"
|
|
// CHECK: @[[STR2:.*]] = {{.*}} c"_ZN1X4var1E\00"
|
|
|
|
// LNX-LABEL: define {{.*}}@_Z4fun1v()
|
|
// MSVC-LABEL: define {{.*}} @"?fun1@@YAPEBDXZ"()
|
|
// CHECK: ret i8* getelementptr inbounds ({{.*}} @[[STR1]], i64 0, i64 0)
|
|
const char *fun1() {
|
|
return __builtin_get_device_side_mangled_name(X::kern1);
|
|
}
|
|
|
|
// LNX-LABEL: define {{.*}}@_Z4fun2v()
|
|
// MSVC-LABEL: define {{.*}}@"?fun2@@YAPEBDXZ"()
|
|
// CHECK: ret i8* getelementptr inbounds ({{.*}} @[[STR2]], i64 0, i64 0)
|
|
__host__ __device__ const char *fun2() {
|
|
return __builtin_get_device_side_mangled_name(X::var1);
|
|
}
|