[IOTDB-4120] Modify build method of LayerMemoryAssigner

apache · Aug 15, 2022 · 5faa9da · 5faa9da
1 parent 02416f4
commit 5faa9da
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/.../src/main/java/org/apache/iotdb/db/mpp/transformation/dag/memory/LayerMemoryAssigner.java b/.../src/main/java/org/apache/iotdb/db/mpp/transformation/dag/memory/LayerMemoryAssigner.java
            : 1);
  }

  public void build() {
     int memoryPartitions = 0;
     for (Entry<Expression, Integer> expressionReferenceEntry :
         expressionReferenceCount.entrySet()) {
-      memoryPartitions +=
-          expressionReferenceEntry.getValue()
-              * (expressionReferenceEntry.getKey() instanceof FunctionExpression ? 2 : 1);
+      Expression expression = expressionReferenceEntry.getKey();
+      if (expression instanceof FunctionExpression) {
+        memoryPartitions += 2;
+      } else if (!(expression instanceof ConstantOperand)) {
+        memoryPartitions += 1;
+      }
     }
     memoryBudgetForSingleReference =
         memoryPartitions == 0 ? memoryBudgetInMB : memoryBudgetInMB / memoryPartitions;
  }

  public int getReference(Expression expression) {