When lowering linalg to call, also link the parent air.herd to the fu…

…nc declaration
Xilinx · Feb 24, 2025 · 7d91fa0 · 7d91fa0
1 parent f2e9e80
commit 7d91fa0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mlir/lib/Conversion/AIRToAIEPass.cpp b/mlir/lib/Conversion/AIRToAIEPass.cpp
@@ -3679,6 +3679,12 @@ class AIRLinalgOpToLibraryCallRewrite
         op, fnNameAttr.getValue(), TypeRange(),
         ValueRange(ArrayRef<Value>(libFnOperands)));
 
+    if (auto herd = op->getParentOfType<air::HerdOp>())
+      rewriter.modifyOpInPlace(herd, [&]() {
+        herd->setAttr("link_with",
+                      StringAttr::get(rewriter.getContext(), linkWith));
+      });
+
     return success();
   }