mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-26 22:46:06 +00:00
11d3e31c60
MSVC and Itanium mangling use different mangling numbers for function-scope structs, which causes inconsistent mangled kernel names in device and host compilations. This patch uses Itanium mangling number for structs in for mangling device side names in CUDA/HIP host compilation on Windows to fix this issue. A state is added to ASTContext to indicate whether the current name mangling is for device side names in host compilation. Device and host mangling number are encoded/decoded as upper and lower half of 32 bit unsigned integer to fit into the original mangling number field for AST. Diagnostic will be emitted if a manglining number exceeds limit. Reviewed by: Artem Belevich, Reid Kleckner Differential Revision: https://reviews.llvm.org/D122734 Fixes: SWDEV-328515
69 lines
1.8 KiB
Plaintext
69 lines
1.8 KiB
Plaintext
// RUN: %clang_cc1 -emit-llvm -o - -aux-triple x86_64-pc-windows-msvc \
|
|
// RUN: -fms-extensions -triple amdgcn-amd-amdhsa \
|
|
// RUN: -target-cpu gfx1030 -fcuda-is-device -x hip %s \
|
|
// RUN: | FileCheck -check-prefix=DEV %s
|
|
|
|
// RUN: %clang_cc1 -emit-llvm -o - -triple x86_64-pc-windows-msvc \
|
|
// RUN: -fms-extensions -aux-triple amdgcn-amd-amdhsa \
|
|
// RUN: -aux-target-cpu gfx1030 -x hip %s \
|
|
// RUN: | FileCheck -check-prefix=HOST %s
|
|
|
|
// RUN: %clang_cc1 -emit-llvm -o - -triple x86_64-pc-windows-msvc \
|
|
// RUN: -fms-extensions -aux-triple amdgcn-amd-amdhsa \
|
|
// RUN: -aux-target-cpu gfx1030 -x hip %s \
|
|
// RUN: | FileCheck -check-prefix=HOST-NEG %s
|
|
|
|
// RUN: %clang_cc1 -emit-llvm -o - -triple x86_64-pc-windows-msvc \
|
|
// RUN: -fms-extensions -x c++ %s \
|
|
// RUN: | FileCheck -check-prefix=CPP %s
|
|
|
|
#if __HIP__
|
|
#include "Inputs/cuda.h"
|
|
#endif
|
|
|
|
// Check local struct 'Op' uses Itanium mangling number instead of MSVC mangling
|
|
// number in device side name mangling. It is the same in device and host
|
|
// compilation.
|
|
|
|
// DEV: define amdgpu_kernel void @_Z6kernelIZN4TestIiE3runEvE2OpEvv(
|
|
|
|
// HOST-DAG: @{{.*}} = {{.*}}c"_Z6kernelIZN4TestIiE3runEvE2OpEvv\00"
|
|
|
|
// HOST-NEG-NOT: @{{.*}} = {{.*}}c"_Z6kernelIZN4TestIiE3runEvE2Op_1Evv\00"
|
|
#if __HIP__
|
|
template<typename T>
|
|
__attribute__((global)) void kernel()
|
|
{
|
|
}
|
|
#endif
|
|
|
|
// Check local struct 'Op' uses MSVC mangling number in host function name mangling.
|
|
// It is the same when compiled as HIP or C++ program.
|
|
|
|
// HOST-DAG: call void @"??$fun@UOp@?2??run@?$Test@H@@QEAAXXZ@@@YAXXZ"()
|
|
// CPP: call void @"??$fun@UOp@?2??run@?$Test@H@@QEAAXXZ@@@YAXXZ"()
|
|
template<typename T>
|
|
void fun()
|
|
{
|
|
}
|
|
|
|
template <typename T>
|
|
class Test {
|
|
public:
|
|
void run()
|
|
{
|
|
struct Op
|
|
{
|
|
};
|
|
#if __HIP__
|
|
kernel<Op><<<1, 1>>>();
|
|
#endif
|
|
fun<Op>();
|
|
}
|
|
};
|
|
|
|
int main() {
|
|
Test<int> A;
|
|
A.run();
|
|
}
|