MLIR add postop tablegen helper (#1783)

EnzymeAD · Mar 7, 2024 · 0a129ae · 0a129ae
1 parent d12e7cd
commit 0a129ae
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/enzyme/Enzyme/MLIR/Implementations/Common.td b/enzyme/Enzyme/MLIR/Implementations/Common.td
@@ -80,9 +80,10 @@ class GlobalExpr<bit uses_primal, bit uses_shadow, string val> : Operation<uses_
  string value = val;
 }
 
-class Inst<string mnemonic, string dialect_> : Operation</*primal*/1, /*shadow*/0> {
+class Inst<string mnemonic, string dialect_, string postop_=""> : Operation</*primal*/1, /*shadow*/0> {
  string name = mnemonic;
  string dialect = dialect_;
+ string postop = postop_;
 }
 
 def Op {

diff --git a/enzyme/tools/enzyme-tblgen/enzyme-tblgen.cpp b/enzyme/tools/enzyme-tblgen/enzyme-tblgen.cpp
@@ -922,6 +922,10 @@ bool handle(const Twine &curIndent, const Twine &argPattern, raw_ostream &os,
  if (isCall || isIntr)
  os << "})";
  os << ")";
+ if (intrinsic == MLIRDerivatives) {
+ auto postop = Def->getValueAsString("postop");
+ os << postop;
+ }
  if (isCall) {
  os << ")";
  }