Adapt to const tblgen main function (#2099)

* Adapt to const tblgen main function * Update enzyme-tblgen.cpp * fix * fix
EnzymeAD · Oct 2, 2024 · b208929 · b208929
1 parent 7425c37
commit b208929
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp b/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp
@@ -1457,7 +1457,7 @@ void rev_call_args(bool forward, Twine argName, const TGPattern &pattern,
   }
 }
 
-void emit_tmp_free(Record *Def, raw_ostream &os, StringRef builder) {
+void emit_tmp_free(const Record *Def, raw_ostream &os, StringRef builder) {
   const auto args = Def->getValueAsListOfStrings("args");
   // allocating tmp variables is optional, return if not required
   if (args.size() == 0)
@@ -1467,7 +1467,7 @@ void emit_tmp_free(Record *Def, raw_ostream &os, StringRef builder) {
   os << "    CreateDealloc(" << builder << ", true_" << allocName << ");\n";
 }
 
-void emit_tmp_creation(Record *Def, raw_ostream &os, StringRef builder) {
+void emit_tmp_creation(const Record *Def, raw_ostream &os, StringRef builder) {
   const auto args = Def->getValueAsListOfStrings("args");
   // allocating tmp variables is optional, return if not required
   if (args.size() == 0)

diff --git a/enzyme/tools/enzyme-tblgen/enzyme-tblgen.cpp b/enzyme/tools/enzyme-tblgen/enzyme-tblgen.cpp
@@ -2746,7 +2746,12 @@ void emitDiffUse(const RecordKeeper &recordKeeper, raw_ostream &os,
 
 void emitMLIRDerivatives(RecordKeeper &records, raw_ostream &os);
 
-static bool EnzymeTableGenMain(raw_ostream &os, RecordKeeper &records) {
+#if LLVM_VERSION_MAJOR >= 20
+static bool EnzymeTableGenMain(raw_ostream &os, const RecordKeeper &records)
+#else
+static bool EnzymeTableGenMain(raw_ostream &os, RecordKeeper &records)
+#endif
+{
   switch (action) {
   case MLIRDerivatives:
   case CallDerivatives: