File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed
spark/src/main/scala/org/apache/comet Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Original file line number Diff line number Diff line change @@ -93,6 +93,14 @@ class CometExecIterator(
9393 }
9494 val protobufSparkConfigs = builder.build().toByteArray
9595
96+ val memoryLimitPerTask = if (offHeapMode) {
97+ // this per-task limit is not used in native code when using unified memory
98+ // so we can skip calculating it and avoid logging irrelevant information
99+ 0
100+ } else {
101+ getMemoryLimitPerTask(conf)
102+ }
103+
96104 nativeLib.createPlan(
97105 id,
98106 cometBatchIterators,
@@ -107,7 +115,7 @@ class CometExecIterator(
107115 offHeapMode,
108116 memoryPoolType = COMET_EXEC_MEMORY_POOL_TYPE .get(),
109117 memoryLimit,
110- memoryLimitPerTask = getMemoryLimitPerTask(conf) ,
118+ memoryLimitPerTask,
111119 taskAttemptId = TaskContext .get().taskAttemptId,
112120 debug = COMET_DEBUG_ENABLED .get(),
113121 explain = COMET_EXPLAIN_NATIVE_ENABLED .get(),
You can’t perform that action at this time.
0 commit comments