Recursive inside dispatch region function

tenstorrent · Sep 26, 2024 · 4bea10c · 4bea10c
1 parent d7638b1
commit 4bea10c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/Dialect/TTKernel/IR/TTKernelOps.cpp b/lib/Dialect/TTKernel/IR/TTKernelOps.cpp
@@ -19,19 +19,19 @@ namespace mlir::tt::ttkernel {
 static bool insideDispatchOpRegion(mlir::Operation *op) {
   mlir::Operation *parentOp = op->getParentOp();
 
-  if (dyn_cast_or_null<ttmetal::DispatchOp>(parentOp)) {
-    return true;
+  if (!parentOp) {
+    return false;
   }
 
-  if (dyn_cast_or_null<scf::ForOp>(parentOp)) {
+  if (dyn_cast_or_null<ttmetal::DispatchOp>(parentOp)) {
     return true;
   }
 
   if (dyn_cast_or_null<func::FuncOp>(parentOp) &&
       dyn_cast_or_null<mlir::ModuleOp>(parentOp->getParentOp())) {
     return true;
   }
-  return false;
+  return insideDispatchOpRegion(parentOp);
 }
 
 ::mlir::LogicalResult BuiltinOp::verify() {