stdio.h

Go to the documentation of this file.

//===-- 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
//
//===----------------------------------------------------------------------===//
 
#ifndef __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
#define __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__
 
#if !defined(_OPENMP) && !defined(__HIP__) && !defined(__CUDA__)
#error "This file is for GPU offloading compilation only"
#endif
 
#include_next <stdio.h>
 
#if defined(__HIP__) || defined(__CUDA__)
#define __LIBC_ATTRS __attribute__((device))
#else
#define __LIBC_ATTRS
#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
 
__LIBC_ATTRS extern FILE *stderr;
__LIBC_ATTRS extern FILE *stdin;
__LIBC_ATTRS extern FILE *stdout;
 
#pragma omp end declare target
 
// Restore the original macros when compiling on the host.
#if !defined(__NVPTX__) && !defined(__AMDGPU__)
#pragma pop_macro("stderr")
#pragma pop_macro("stdin")
#pragma pop_macro("stdout")
#endif
 
#undef __LIBC_ATTRS
 
#endif // __CLANG_LLVM_LIBC_WRAPPERS_STDIO_H__