From 95a17684fa6d3155380227bd1c775d4eccb4078b Mon Sep 17 00:00:00 2001 From: Xinsheng Chen Date: Tue, 18 Mar 2025 16:04:39 +0800 Subject: [PATCH] reduce memory alloc (in certain cases) --- python/jittor/src/mem/allocator/sfrl_allocator.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/jittor/src/mem/allocator/sfrl_allocator.cc b/python/jittor/src/mem/allocator/sfrl_allocator.cc index add3aef9..22bf89b8 100644 --- a/python/jittor/src/mem/allocator/sfrl_allocator.cc +++ b/python/jittor/src/mem/allocator/sfrl_allocator.cc @@ -15,8 +15,8 @@ namespace jittor { DEFINE_FLAG(int, use_sfrl_allocator, 1, "Enable sfrl allocator"); -DEFINE_FLAG(int64, sfrl_large_block_size_device, 20971520, "sfrl_large_block_size, larger will reduce memory shard, only affect device"); -constexpr int64 sfrl_large_block_size_cpu=20971520; +DEFINE_FLAG(int64, sfrl_large_block_size_device, 5242880, "sfrl_large_block_size, larger will reduce memory shard, only affect device"); +constexpr int64 sfrl_large_block_size_cpu=5242880; std::vector CachingBlockPool::block_ids; //start from 1