Skip external consts when walking operands

Xilinx · Oct 4, 2024 · a760d41 · a760d41
1 parent 5793d27
commit a760d41
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/lib/Transform/XTenMinimizeLiveTensors.cpp b/lib/Transform/XTenMinimizeLiveTensors.cpp
@@ -192,14 +192,25 @@ FailureOr<SmallVector<Value>> getFmOperands(Operation *op) {
   if (isa<func::FuncOp>(op))
     return {{}};
 
+  const auto filterOutExternalConst = [](SmallVector<Value> operands) {
+    SmallVector<Value> filteredOperands;
+    for (const auto operand : operands) {
+      if (!isa_and_nonnull<amd::xten_nn::LoadExternalConstOp>(
+              operand.getDefiningOp())) {
+        filteredOperands.push_back(operand);
+      }
+    }
+    return filteredOperands;
+  };
+
   if (isInCoreChain(op))
     return {getSubgraphIFMs(op)};
 
   if (isConcatSubgraph(op))
     return {getSubgraphIFMs(op)};
 
   if (isTemplatedGraph(op))
-    return {op->getOperands()};
+    return {filterOutExternalConst(op->getOperands())};
 
   // Otherwise, this is a PseudoOp and IFM is the first operand.
   if (!(isAnyPseudoOp(op) || isInterfaceOp(op))) {
@@ -225,7 +236,8 @@ size_t getSize(Value val) {
 
   if (auto complexType = elementType.dyn_cast<ComplexType>()) {
     elementType = complexType.getElementType();
-    return (elementType.getIntOrFloatBitWidth() * type.getNumElements() * 2) / 8;
+    return (elementType.getIntOrFloatBitWidth() * type.getNumElements() * 2) /
+           8;
   }
   llvm_unreachable("Does not know how to compute size");
 }
@@ -299,7 +311,8 @@ class XTenMinimizeLiveTensorsPass
       } else {
         fmResults = SmallVector<Value>(currFn.getBody().front().getArguments());
       }
-      std::optional<Value> const sharesResultMemory = sharesMemoryWithResult(defOp);
+      std::optional<Value> const sharesResultMemory =
+          sharesMemoryWithResult(defOp);
       OpInfo info = {.op = defOp,
                      .operands = *fmOperands,
                      .results = fmResults,