clang 18.0.0git
stdio.h
Go to the documentation of this file.
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__