80 lines
2.9 KiB
C
80 lines
2.9 KiB
C
//===-- Wrapper for C standard stdio.h declarations on the GPU ------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#if !defined(_OPENMP) && !defined(__HIP__) && !defined(__CUDA__)
|
|
#error "This file is for GPU offloading compilation only"
|
|
#endif
|
|
|
|
#include_next <stdio.h>
|
|
|
|
// In some old versions of glibc, other standard headers sometimes define
|
|
// special macros (e.g., __need_FILE) before including stdio.h to cause stdio.h
|
|
// to produce special definitions. Future includes of stdio.h when those
|
|
// special macros are undefined are expected to produce the normal definitions
|
|
// from stdio.h.
|
|
//
|
|
// We do not apply our include guard (__CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__)
|
|
// unconditionally to the above include_next. Otherwise, after an occurrence of
|
|
// the first glibc stdio.h use case described above, the include_next would be
|
|
// skipped for remaining includes of stdio.h, leaving required symbols
|
|
// undefined.
|
|
//
|
|
// We make the following assumptions to handle all use cases:
|
|
//
|
|
// 1. If the above include_next produces special glibc definitions, then (a) it
|
|
// does not produce the normal definitions that we must intercept below, (b)
|
|
// the current file was included from a glibc header that already defined
|
|
// __GLIBC__ (usually by including glibc's <features.h>), and (c) the above
|
|
// include_next does not define _STDIO_H. In that case, we skip the rest of
|
|
// the current file and don't guard against future includes.
|
|
// 2. If the above include_next produces the normal stdio.h definitions, then
|
|
// either (a) __GLIBC__ is not defined because C headers are from some other
|
|
// libc implementation or (b) the above include_next defines _STDIO_H to
|
|
// prevent the above include_next from having any effect in the future.
|
|
#if !defined(__GLIBC__) || defined(_STDIO_H)
|
|
|
|
#ifndef __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
|
|
#define __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
|
|
|
|
#if __has_include(<llvm-libc-decls/stdio.h>)
|
|
|
|
#if defined(__HIP__) || defined(__CUDA__)
|
|
#define __LIBC_ATTRS __attribute__((device))
|
|
#endif
|
|
|
|
// Some headers provide these as macros. Temporarily undefine them so they do
|
|
// not conflict with any definitions for the GPU.
|
|
|
|
#pragma push_macro("stdout")
|
|
#pragma push_macro("stdin")
|
|
#pragma push_macro("stderr")
|
|
|
|
#undef stdout
|
|
#undef stderr
|
|
#undef stdin
|
|
|
|
#pragma omp begin declare target
|
|
|
|
#include <llvm-libc-decls/stdio.h>
|
|
|
|
#pragma omp end declare target
|
|
|
|
#undef __LIBC_ATTRS
|
|
|
|
// Restore the original macros when compiling on the host.
|
|
#if !defined(__NVPTX__) && !defined(__AMDGPU__)
|
|
#pragma pop_macro("stdout")
|
|
#pragma pop_macro("stderr")
|
|
#pragma pop_macro("stdin")
|
|
#endif
|
|
|
|
#endif
|
|
|
|
#endif // __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
|
|
|
|
#endif
|