mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-25 12:26:07 +00:00
d4216b5d0b
This enables the AMDGPU specific implementation of `printf` when compiling for AMDGCN flavoured SPIR-V, the consequence being that the expansion into ROCDL calls & friends gets expanded before "lowering" to SPIR-V and gets carried through. The only relatively "novel" aspect is that the `callAppendStringN` is simplified to take the type of the passed in arguments, as opposed to querying them from the module. This is a neutral change since the arguments were passed directly to the call, without any attempt to cast them, hence the assumption that the actual types match the formal ones was already baked in.
32 lines
1.6 KiB
Plaintext
32 lines
1.6 KiB
Plaintext
// REQUIRES: amdgpu-registered-target
|
|
// RUN: %clang_cc1 -triple amdgcn-amd-amdhsa -emit-llvm -disable-llvm-optzns -mprintf-kind=hostcall -fno-builtin-printf -fcuda-is-device \
|
|
// RUN: -o - %s | FileCheck --check-prefixes=CHECK,HOSTCALL %s
|
|
// RUN: %clang_cc1 -triple spirv64-amd-amdhsa -emit-llvm -disable-llvm-optzns -mprintf-kind=hostcall -fno-builtin-printf -fcuda-is-device \
|
|
// RUN: -o - %s | FileCheck --check-prefixes=CHECK-AMDGCNSPIRV,HOSTCALL-AMDGCNSPIRV %s
|
|
// RUN: %clang_cc1 -triple amdgcn-amd-amdhsa -emit-llvm -disable-llvm-optzns -mprintf-kind=buffered -fno-builtin-printf -fcuda-is-device \
|
|
// RUN: -o - %s | FileCheck --check-prefixes=CHECK,BUFFERED %s
|
|
// RUN: %clang_cc1 -triple spirv64-amd-amdhsa -emit-llvm -disable-llvm-optzns -mprintf-kind=buffered -fno-builtin-printf -fcuda-is-device \
|
|
// RUN: -o - %s | FileCheck --check-prefixes=CHECK-AMDGCNSPIRV,BUFFERED-AMDGCNSPIRV %s
|
|
|
|
#define __device__ __attribute__((device))
|
|
|
|
extern "C" __device__ int printf(const char *format, ...);
|
|
|
|
// CHECK-LABEL: @_Z4foo1v()
|
|
__device__ int foo1() {
|
|
// HOSTCALL: call i64 @__ockl_printf_begin
|
|
// HOSTCALL-AMDGCNSPIRV: call addrspace(4) i64 @__ockl_printf_begin
|
|
// BUFFERED: call ptr addrspace(1) @__printf_alloc
|
|
// BUFFERED-AMDGCNSPIRV: call addrspace(4) ptr addrspace(1) @__printf_alloc
|
|
// CHECK-NOT: call i32 (ptr, ...) @printf
|
|
// CHECK-AMDGCNSPIRV-NOT: call i32 (ptr, ...) @printf
|
|
return __builtin_printf("Hello World\n");
|
|
}
|
|
|
|
// CHECK-LABEL: @_Z4foo2v()
|
|
__device__ int foo2() {
|
|
// CHECK: call i32 (ptr, ...) @printf
|
|
// CHECK-AMDGCNSPIRV: call spir_func addrspace(4) i32 (ptr addrspace(4), ...) @printf
|
|
return printf("Hello World\n");
|
|
}
|