fixup

apache · Nov 10, 2023 · ab65468 · ab65468
1 parent 28592f2
commit ab65468
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gluten-core/src/main/java/io/glutenproject/memory/memtarget/spark/TreeMemoryConsumers.java b/gluten-core/src/main/java/io/glutenproject/memory/memtarget/spark/TreeMemoryConsumers.java
@@ -56,7 +56,7 @@ private TreeMemoryTarget getSharedAccount(TaskMemoryManager tmm) {
       synchronized (MAP) {
         return MAP.computeIfAbsent(tmm, m -> {
           TreeMemoryTarget tmc = new TreeMemoryConsumer(m);
-          return tmc.newChild("root", GlutenConfig.getConf().conservativeTaskOffHeapMemorySize(), Spiller.NO_OP, Collections.emptyMap());
+          return tmc.newChild("root", perTaskCapacity, Spiller.NO_OP, Collections.emptyMap());
         });
       }
     }