clang/lib/Headers/llvm_libc_wrappers/stdio.h

   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__