654b76321a
This patch adds a call to the CUFInit function just after `ProgramStart` when CUDA Fortran is enabled to initialize the CUDA context. This allows us to set up some context information like the stack limit that can be defined by an environment variable `ACC_OFFLOAD_STACKSIZE=<value>`.
71 lines
2.2 KiB
C++
71 lines
2.2 KiB
C++
//===-- runtime/environment.h -----------------------------------*- C++ -*-===//
|
|
//
|
|
// 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 FORTRAN_RUNTIME_ENVIRONMENT_H_
|
|
#define FORTRAN_RUNTIME_ENVIRONMENT_H_
|
|
|
|
#include "flang/Common/optional.h"
|
|
#include "flang/Decimal/decimal.h"
|
|
|
|
struct EnvironmentDefaultList;
|
|
|
|
namespace Fortran::runtime {
|
|
|
|
class Terminator;
|
|
|
|
RT_OFFLOAD_VAR_GROUP_BEGIN
|
|
#if FLANG_BIG_ENDIAN
|
|
constexpr bool isHostLittleEndian{false};
|
|
#elif FLANG_LITTLE_ENDIAN
|
|
constexpr bool isHostLittleEndian{true};
|
|
#else
|
|
#error host endianness is not known
|
|
#endif
|
|
RT_OFFLOAD_VAR_GROUP_END
|
|
|
|
// External unformatted I/O data conversions
|
|
enum class Convert { Unknown, Native, LittleEndian, BigEndian, Swap };
|
|
|
|
RT_API_ATTRS Fortran::common::optional<Convert> GetConvertFromString(
|
|
const char *, std::size_t);
|
|
|
|
struct ExecutionEnvironment {
|
|
#if !defined(_OPENMP)
|
|
// FIXME: https://github.com/llvm/llvm-project/issues/84942
|
|
constexpr
|
|
#endif
|
|
ExecutionEnvironment(){};
|
|
void Configure(int argc, const char *argv[], const char *envp[],
|
|
const EnvironmentDefaultList *envDefaults);
|
|
const char *GetEnv(
|
|
const char *name, std::size_t name_length, const Terminator &terminator);
|
|
|
|
int argc{0};
|
|
const char **argv{nullptr};
|
|
char **envp{nullptr};
|
|
|
|
int listDirectedOutputLineLengthLimit{79}; // FORT_FMT_RECL
|
|
enum decimal::FortranRounding defaultOutputRoundingMode{
|
|
decimal::FortranRounding::RoundNearest}; // RP(==PN)
|
|
Convert conversion{Convert::Unknown}; // FORT_CONVERT
|
|
bool noStopMessage{false}; // NO_STOP_MESSAGE=1 inhibits "Fortran STOP"
|
|
bool defaultUTF8{false}; // DEFAULT_UTF8
|
|
bool checkPointerDeallocation{true}; // FORT_CHECK_POINTER_DEALLOCATION
|
|
|
|
// CUDA related variables
|
|
std::size_t cudaStackLimit{0}; // ACC_OFFLOAD_STACK_SIZE
|
|
};
|
|
|
|
RT_OFFLOAD_VAR_GROUP_BEGIN
|
|
extern RT_VAR_ATTRS ExecutionEnvironment executionEnvironment;
|
|
RT_OFFLOAD_VAR_GROUP_END
|
|
|
|
} // namespace Fortran::runtime
|
|
|
|
#endif // FORTRAN_RUNTIME_ENVIRONMENT_H_
|