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