Add another type of include header

enzyme/Enzyme/Clang/include_utils.td

@@ -1,9 +1,14 @@
-class Headers<string filename_, string contents_> {
+class InlineHeader<string filename_, string contents_> {
  string filename = filename_;
  string contents = contents_;
 }
 
-def : Headers<"/enzymeroot/enzyme/utils", [{
+class FileHeader<string filename_out_, string filename_in_> {
+ string filename_out = filename_out_;
+ string filename_in = filename_in_;
+}
+
+def : InlineHeader<"/enzymeroot/enzyme/utils", [{
 #pragma once
 
 extern int enzyme_dup;
@@ -263,7 +268,7 @@ namespace enzyme {
 }
 }]>;
 
-def : Headers<"/enzymeroot/enzyme/type_traits", [{
+def : InlineHeader<"/enzymeroot/enzyme/type_traits", [{
 #pragma once
 
 #include <type_traits>
@@ -312,7 +317,7 @@ namespace impl {
 }
 }]>;
 
-def : Headers<"/enzymeroot/enzyme/tuple", [{
+def : InlineHeader<"/enzymeroot/enzyme/tuple", [{
 #pragma once
 
 /////////////
@@ -449,10 +454,12 @@ constexpr auto tuple_cat(Tuples&&... tuples) {
 #undef _NOEXCEPT
 }]>;
 
-def : Headers<"/enzymeroot/enzyme/enzyme", [{
+def : InlineHeader<"/enzymeroot/enzyme/enzyme", [{
 #ifdef __cplusplus
 #include "enzyme/utils"
 #else
 #warning "Enzyme wrapper templates only available in C++"
 #endif
 }]>;
+
+def : FileHeader<"/enzymeroot/enzyme/mpfr", "Runtime/MPFR.cpp">;

enzyme/tools/enzyme-tblgen/enzyme-tblgen.cpp

@@ -16,6 +16,7 @@
 #include "llvm/ADT/StringSet.h"
 #include "llvm/Support/CommandLine.h"
 #include "llvm/Support/FormatVariadic.h"
+#include "llvm/Support/Path.h"
 #include "llvm/Support/PrettyStackTrace.h"
 #include "llvm/Support/Signals.h"
 #include "llvm/TableGen/Error.h"
@@ -1252,18 +1253,38 @@ void printDiffUse(
 
 static void emitHeaderIncludes(const RecordKeeper &recordKeeper,
  raw_ostream &os) {
- const auto &patterns = recordKeeper.getAllDerivedDefinitions("Headers");
  os << "const char* include_headers[][2] = {\n";
  bool seen = false;
- for (Record *pattern : patterns) {
- if (seen)
- os << ",\n";
- auto filename = pattern->getValueAsString("filename");
- auto contents = pattern->getValueAsString("contents");
- os << "{\"" << filename << "\"\n,";
- os << "R\"(" << contents << ")\"\n";
- os << "}";
- seen = true;
+ {
+ const auto &patterns = recordKeeper.getAllDerivedDefinitions("InlineHeader");
+ for (Record *pattern : patterns) {
+ if (seen)
+ os << ",\n";
+ auto filename = pattern->getValueAsString("filename");
+ auto contents = pattern->getValueAsString("contents");
+ os << "{\"" << filename << "\"\n,";
+ os << "R\"(" << contents << ")\"\n";
+ os << "}";
+ seen = true;
+ }
+ }
+ {
+ const auto &patterns = recordKeeper.getAllDerivedDefinitions("FileHeader");
+ for (Record *pattern : patterns) {
+ if (seen)
+ os << ",\n";
+ auto filename_out = pattern->getValueAsString("filename_out");
+ std::string filename_in = pattern->getValueAsString("filename_in").str();
+ std::string included_file;
+ auto contents = llvm::SrcMgr.OpenIncludeFile(filename_in, included_file);
+ //llvm::MemoryBuffer::getFile(filename_in, /*IsText=*/true);
+ if (!contents)
+ PrintFatalError(pattern->getLoc(), Twine("Could not read file ") + filename_in);
+ os << "{\"" << filename_out << "\"\n,";
+ os << "R\"(" << contents.get()->getBuffer() << ")\"\n";
+ os << "}";
+ seen = true;
+ }
  }
  os << "};\n";
 }