Fmt 1

Author: jessicacotturone21

src/enzyme_ad/jax/Dialect/Tessera/Ops.cpp

@@ -4,9 +4,9 @@
 #include "Dialect.h"
 #include "Ops.h"
 #include "mlir/IR/IRMapping.h"
+#include "mlir/Interfaces/CallInterfaces.h"
 #include "mlir/Interfaces/FunctionImplementation.h"
 #include "mlir/Interfaces/FunctionInterfaces.h"
-#include "mlir/Interfaces/CallInterfaces.h"
 #include "mlir/Interfaces/SideEffectInterfaces.h"
 
 using namespace mlir;

src/enzyme_ad/jax/Dialect/Tessera/Ops.h

@@ -1,13 +1,13 @@
 #ifndef ENZYME_AD_JAX_TESSERA_OPS_H
 #define ENZYME_AD_JAX_TESSERA_OPS_H
 
-#include "mlir/IR/OpDefinition.h"
 #include "mlir/IR/BuiltinOps.h"
 #include "mlir/IR/Dialect.h"
-#include "mlir/Interfaces/FunctionInterfaces.h"
+#include "mlir/IR/OpDefinition.h"
 #include "mlir/Interfaces/CallInterfaces.h"
-#include "mlir/Interfaces/SideEffectInterfaces.h"
 #include "mlir/Interfaces/ControlFlowInterfaces.h"
+#include "mlir/Interfaces/FunctionInterfaces.h"
+#include "mlir/Interfaces/SideEffectInterfaces.h"
 
 #define GET_OP_CLASSES
 #include "src/enzyme_ad/jax/Dialect/Tessera/TesseraOps.h.inc"

src/enzyme_ad/jax/Passes/Tessera/FuncToTessera.cpp

@@ -46,7 +46,7 @@ class FuncOpRewrite final : public OpRewritePattern<func::FuncOp> {
     // Create the `tessera.define` op
     auto tesseraDefineOp = rewriter.create<tessera::DefineOp>(
       funcOp.getLoc(), funcOp.getName(), fnType);
-    
+
     // Copy over all attributes other than the function name and type.
     for (const auto &namedAttr : funcOp->getAttrs()) {
       if (namedAttr.getName() != funcOp.getFunctionTypeAttrName() &&
@@ -58,8 +58,7 @@ class FuncOpRewrite final : public OpRewritePattern<func::FuncOp> {
     if (!funcOp.isExternal()) {
       IRMapping mapper;
       funcOp.getBody().cloneInto(&tesseraDefineOp.getBody(),
-                            tesseraDefineOp.getBody().end(),
-                            mapper);
+                                 tesseraDefineOp.getBody().end(), mapper);
 
       // Now walk through the cloned operations and convert func.return to
       // tessera.return
@@ -149,23 +148,24 @@ struct FuncToTesseraPass
   }
 
   void runOnOperation() override {
-  MLIRContext *ctx = &getContext();
+    MLIRContext *ctx = &getContext();
 
-  ConversionTarget target(*ctx);
-  target.addLegalDialect<tessera::TesseraDialect>();
-  target.addLegalDialect<BuiltinDialect>();
-  target.addIllegalDialect<func::FuncDialect>();
+    ConversionTarget target(*ctx);
+    target.addLegalDialect<tessera::TesseraDialect>();
+    target.addLegalDialect<BuiltinDialect>();
+    target.addIllegalDialect<func::FuncDialect>();
 
-  RewritePatternSet patterns(ctx);
+    RewritePatternSet patterns(ctx);
 
-  patterns.add<FuncOpRewrite, CallOpRewrite, ReturnOpRewrite>(ctx);
+    patterns.add<FuncOpRewrite, CallOpRewrite, ReturnOpRewrite>(ctx);
 
-  if (failed(applyFullConversion(getOperation(), target, std::move(patterns))))
-    signalPassFailure();
-}
+    if (failed(
+            applyFullConversion(getOperation(), target, std::move(patterns))))
+      signalPassFailure();
+  }
 };
 
 std::unique_ptr<mlir::Pass> createFuncToTesseraPass() {
   return std::make_unique<FuncToTesseraPass>();
 }
-}
+} // namespace mlir::enzyme::tessera

src/enzyme_ad/jax/Passes/Tessera/Passes.h

@@ -1,8 +1,8 @@
 #ifndef TESSERA_PASSES_H
 #define TESSERA_PASSES_H
 
-#include "src/enzyme_ad/jax/Dialect/Tessera/Ops.h"
 #include "mlir/Pass/Pass.h"
+#include "src/enzyme_ad/jax/Dialect/Tessera/Ops.h"
 
 namespace mlir {
 namespace enzyme {

src/enzyme_ad/jax/Passes/Tessera/TesseraToFunc.cpp

@@ -6,18 +6,18 @@
 //===----------------------------------------------------------------------===//
 
 #include "mlir/Bytecode/BytecodeOpInterface.h"
-#include "src/enzyme_ad/jax/Dialect/Tessera/Ops.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
-#include "mlir/IR/PatternMatch.h"
+#include "mlir/IR/BuiltinOps.h"
 #include "mlir/IR/IRMapping.h"
-#include "mlir/Transforms/DialectConversion.h"
-#include "mlir/Interfaces/FunctionInterfaces.h"
+#include "mlir/IR/PatternMatch.h"
 #include "mlir/Interfaces/CallInterfaces.h"
+#include "mlir/Interfaces/FunctionInterfaces.h"
 #include "mlir/Pass/Pass.h"
-#include "mlir/IR/BuiltinOps.h"
+#include "mlir/Transforms/DialectConversion.h"
+#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
 #include "src/enzyme_ad/jax/Dialect/Tessera/Dialect.h"
+#include "src/enzyme_ad/jax/Dialect/Tessera/Ops.h"
 #include "src/enzyme_ad/jax/Passes/Tessera/Passes.h"
-#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
 
 using namespace mlir;
 using namespace mlir::enzyme;
@@ -57,9 +57,8 @@ class DefineOpRewrite final : public OpRewritePattern<tessera::DefineOp> {
     // Clone body of function
     if (!defineOp.isExternal()) {
       IRMapping mapper;
-      defineOp.getBody().cloneInto(&funcOp.getBody(),
-                            funcOp.getBody().end(),
-                            mapper);
+      defineOp.getBody().cloneInto(&funcOp.getBody(), funcOp.getBody().end(),
+                                   mapper);
     }
 
     rewriter.eraseOp(defineOp);
@@ -120,18 +119,18 @@ struct TesseraToFuncPass
   }
 
   void runOnOperation() override {
-  MLIRContext *ctx = &getContext();
-  RewritePatternSet patterns(ctx);
+    MLIRContext *ctx = &getContext();
+    RewritePatternSet patterns(ctx);
 
-  patterns.add<DefineOpRewrite, CallOpRewrite, ReturnOpRewrite>(ctx);
+    patterns.add<DefineOpRewrite, CallOpRewrite, ReturnOpRewrite>(ctx);
 
-  if (failed(applyPatternsAndFoldGreedily(getOperation(),
-                                          std::move(patterns))))
-    signalPassFailure();
-}
+    if (failed(
+            applyPatternsAndFoldGreedily(getOperation(), std::move(patterns))))
+      signalPassFailure();
+  }
 };
 
 std::unique_ptr<mlir::Pass> createTesseraToFuncPass() {
   return std::make_unique<TesseraToFuncPass>();
 }
-}
+} // namespace mlir::enzyme::tessera