diff --git a/flang/lib/Frontend/CompilerInvocation.cpp b/flang/lib/Frontend/CompilerInvocation.cpp --- a/flang/lib/Frontend/CompilerInvocation.cpp +++ b/flang/lib/Frontend/CompilerInvocation.cpp @@ -8,6 +8,7 @@ #include "flang/Frontend/CompilerInvocation.h" #include "flang/Frontend/PreprocessorOptions.h" +#include "flang/Version.inc" #include "clang/Basic/AllDiagnostics.h" #include "clang/Basic/DiagnosticDriver.h" #include "clang/Basic/DiagnosticOptions.h" @@ -258,6 +259,18 @@ std::vector searchDirectories{"."s}; fortranOptions.searchDirectories = searchDirectories; fortranOptions.isFixedForm = false; + + // Populate macro list with version numbers and other predefinitions. + fortranOptions.predefinitions.emplace_back("__flang__", "1"); + fortranOptions.predefinitions.emplace_back( + "__flang_major__", FLANG_VERSION_MAJOR_STRING); + fortranOptions.predefinitions.emplace_back( + "__flang_minor__", FLANG_VERSION_MINOR_STRING); + fortranOptions.predefinitions.emplace_back( + "__flang_patchlevel__", FLANG_VERSION_PATCHLEVEL_STRING); +#if __x86_64__ + fortranOptions.predefinitions.emplace_back("__x86_64__", "1"); +#endif } void CompilerInvocation::setFortranOpts() { diff --git a/flang/test/Flang-Driver/compiler-defined-macros.f90 b/flang/test/Flang-Driver/compiler-defined-macros.f90 new file mode 100644 --- /dev/null +++ b/flang/test/Flang-Driver/compiler-defined-macros.f90 @@ -0,0 +1,24 @@ +! Check that the driver supports the standard version predefinitions that f18 does. + +! REQUIRES: new-flang-driver + +!-------------------------- +! FLANG DRIVER (flang-new) +!-------------------------- +! RUN: %flang-new -E %s 2>&1 | FileCheck %s --ignore-case + +!----------------------------------------- +! FRONTEND FLANG DRIVER (flang-new -fc1) +!----------------------------------------- +! RUN: %flang-new -fc1 -E %s 2>&1 | FileCheck %s --ignore-case + +!----------------- +! EXPECTED OUTPUT +!----------------- +! CHECK: flang_major = {{[1-9][0-9]*$}} +! CHECK: flang_minor = {{[0-9]+$}} +! CHECK: flang_patchlevel = {{[0-9]+$}} + +integer, parameter :: flang_major = __flang_major__ +integer, parameter :: flang_minor = __flang_minor__ +integer, parameter :: flang_patchlevel = __flang_patchlevel__