This is an automated email from the ASF dual-hosted git repository. binjieyang pushed a commit to branch CELEBORN-1463 in repository https://gitbox.apache.org/repos/asf/celeborn.git
commit 03d3845c69c8b70b2a1d144fabe6e579258e7cf9 Author: binjie yang <[email protected]> AuthorDate: Sun Jun 16 16:47:03 2024 +0800 [CELEBORN-1463] Create network memory allocator with celeborn.network.memory.allocator.numArenas --- .../main/java/org/apache/celeborn/common/network/util/NettyUtils.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/common/src/main/java/org/apache/celeborn/common/network/util/NettyUtils.java b/common/src/main/java/org/apache/celeborn/common/network/util/NettyUtils.java index 4c941e864..60eb44744 100644 --- a/common/src/main/java/org/apache/celeborn/common/network/util/NettyUtils.java +++ b/common/src/main/java/org/apache/celeborn/common/network/util/NettyUtils.java @@ -150,7 +150,8 @@ public class NettyUtils { allowCache && conf.getCelebornConf().networkMemoryAllocatorAllowCache()); } PooledByteBufAllocator allocator = - createPooledByteBufAllocator(conf.preferDirectBufs(), allowCache, conf.clientThreads()); + createPooledByteBufAllocator( + conf.preferDirectBufs(), allowCache, conf.getCelebornConf().networkAllocatorArenas()); if (source != null) { String poolName = "default-netty-pool"; Map<String, String> labels = new HashMap<>();
