add some silly debugs

tenstorrent · Dec 26, 2024 · ee397ca · ee397ca
1 parent 6e0fdaa
commit ee397ca
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/Dialect/TTNN/Transforms/Passes.cpp b/lib/Dialect/TTNN/Transforms/Passes.cpp
@@ -64,6 +64,7 @@ class TTNNDeallocate : public impl::TTNNDeallocateBase<TTNNDeallocate> {
   }
 
   void runOnOperation() final {
+    llvm::outs() << "s TTNNDeallocate::rOO\n";
     ModuleOp module = getOperation();
     IRRewriter rewriter(&getContext());
 
@@ -115,6 +116,7 @@ class TTNNDeallocate : public impl::TTNNDeallocateBase<TTNNDeallocate> {
         }
       });
     });
+    llvm::outs() << "e TTNNDeallocate::rOO\n";
   }
 };
 
@@ -126,6 +128,8 @@ class TTNNDecomposeLayouts
       TTNNDecomposeLayouts>::TTNNDecomposeLayoutsBase;
 
   void runOnOperation() final {
+    llvm::outs() << "s TTNNDecomposeLayouts::rOO\n";
+
     ModuleOp module = getOperation();
     IRRewriter rewriter(&getContext());
     llvm::SmallVector<Operation *> opsToReplace;
@@ -147,6 +151,7 @@ class TTNNDecomposeLayouts
                                       rewriter);
       rewriter.eraseOp(op);
     }
+    llvm::outs() << "e TTNNDecomposeLayouts::rOO\n";
   }
 
 private:
@@ -906,6 +911,8 @@ class TTNNCreateInputGenerators
       TTNNCreateInputGenerators>::TTNNCreateInputGeneratorsBase;
 
   void runOnOperation() final {
+    llvm::outs() << "s TTNNCreateInputGenerators::rOO\n";
+
     ModuleOp module = getOperation();
     IRRewriter rewriter(&getContext());
 
@@ -1074,6 +1081,8 @@ class TTNNCreateInputGenerators
           rewriter.getI32IntegerAttr(0));
       rewriter.create<func::ReturnOp>(mainFuncOp->getLoc(), constantZero);
     }
+
+    llvm::outs() << "e TTNNCreateInputGenerators::rOO\n";
   }
 };
 
@@ -1086,6 +1095,8 @@ class TTNNModifySignaturesForDylib
       TTNNModifySignaturesForDylib>::TTNNModifySignaturesForDylibBase;
 
   void runOnOperation() final {
+    llvm::outs() << "s TTNNModifySignaturesForDylib::rOO\n";
+
     ModuleOp module = getOperation();
     IRRewriter rewriter(&getContext());
 
@@ -1166,6 +1177,7 @@ class TTNNModifySignaturesForDylib
       //
       entryBlock.eraseArguments(1, originalFuncType.getInputs().size());
     }
+    llvm::outs() << "e TTNNModifySignaturesForDylib::rOO\n";
   }
 };
 

diff --git a/lib/Dialect/TTNN/Transforms/TTNNLayout.cpp b/lib/Dialect/TTNN/Transforms/TTNNLayout.cpp
@@ -472,6 +472,7 @@ class TTNNLayout : public impl::TTNNLayoutBase<TTNNLayout> {
         signalPassFailure();
         return;
       }
+      llvm::outs() << "e TTNNLayout::rOO\n";
     }
   }
-Original file line number
+Diff line change
@@ Expand Up @@
             signalPassFailure();
             return;
           }
+          llvm::outs() << "e TTNNLayout::rOO\n";
         }
       }
@@ Expand Down @@