1 //===-- Wrapper for C standard stdio.h declarations on the GPU ------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #ifndef __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
10 #define __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
11 
12 #if !defined(_OPENMP) && !defined(__HIP__) && !defined(__CUDA__)
13 #error "This file is for GPU offloading compilation only"
14 #endif
15 
16 #include_next <stdio.h>
17 
18 #if __has_include(<llvm-libc-decls/stdio.h>)
19 
20 #if defined(__HIP__) || defined(__CUDA__)
21 #define __LIBC_ATTRS __attribute__((device))
22 #endif
23 
24 // Some headers provide these as macros. Temporarily undefine them so they do
25 // not conflict with any definitions for the GPU.
26 
27 #pragma push_macro("stdout")
28 #pragma push_macro("stdin")
29 #pragma push_macro("stderr")
30 
31 #undef stdout
32 #undef stderr
33 #undef stdin
34 
35 #pragma omp begin declare target
36 
37 #include <llvm-libc-decls/stdio.h>
38 
39 #pragma omp end declare target
40 
41 #undef __LIBC_ATTRS
42 
43 // Restore the original macros when compiling on the host.
44 #if !defined(__NVPTX__) && !defined(__AMDGPU__)
45 #pragma pop_macro("stdout")
46 #pragma pop_macro("stderr")
47 #pragma pop_macro("stdin")
48 #endif
49 
50 #endif
51 
52 #endif // __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
53