Merge branch 'main' into extern-register-pass

EnzymeAD · Feb 7, 2025 · e44e3b7 · e44e3b7
2 parents 77089c9 + e394dbf
commit e44e3b7
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/enzyme/BUILD b/enzyme/BUILD
@@ -250,6 +250,7 @@ td_library(
     srcs = [
         "Enzyme/MLIR/Dialect/Dialect.td",
     ],
+    includes = ["."],
     deps = [
         "@llvm-project//mlir:ControlFlowInterfacesTdFiles",
         "@llvm-project//mlir:FunctionInterfacesTdFiles",

diff --git a/enzyme/Enzyme/MLIR/Implementations/Common.td b/enzyme/Enzyme/MLIR/Implementations/Common.td
@@ -103,10 +103,11 @@ class StaticSelect<string condition_> : Operation</*usesPrimal*/0, /*usesShadow*
   string condition = condition_;
 }
 
-class Inst<string mnemonic, string dialect_, string postop_="">  : Operation</*primal*/1, /*shadow*/0> {
+class Inst<string mnemonic, string dialect_, string postop_="", string preop_="">  : Operation</*primal*/1, /*shadow*/0> {
   string name = mnemonic;
   string dialect = dialect_;
   string postop = postop_;
+  string preop = preop_;
 }
 
 def Op {

diff --git a/enzyme/tools/enzyme-tblgen/enzyme-tblgen.cpp b/enzyme/tools/enzyme-tblgen/enzyme-tblgen.cpp
@@ -1094,6 +1094,10 @@ bool handle(const Twine &curIndent, const Twine &argPattern, raw_ostream &os,
       } else if (opName == "CheckedDiv") {
         os << "checkedDiv(" << builder << ", ";
       } else if (intrinsic == MLIRDerivatives) {
+        if (intrinsic == MLIRDerivatives) {
+          auto preop = Def->getValueAsString("preop");
+          os << preop;
+        }
         auto dialect = Def->getValueAsString("dialect");
         os << builder << ".create<" << dialect << "::" << opName
            << ">(op.getLoc(), ";