From fda155f99935053e107ec2c808b0eaf0e405e312 Mon Sep 17 00:00:00 2001 From: Exusial Date: Wed, 26 Feb 2025 16:21:30 +0800 Subject: [PATCH] debug. --- python/jittor/src/mem/allocator/cuda_dual_allocator.h | 2 +- python/jittor/src/mem/allocator/sfrl_allocator.h | 2 +- python/jittor/src/mem/allocator/temp_allocator.h | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/jittor/src/mem/allocator/cuda_dual_allocator.h b/python/jittor/src/mem/allocator/cuda_dual_allocator.h index 0cc94a31..b0be66c8 100644 --- a/python/jittor/src/mem/allocator/cuda_dual_allocator.h +++ b/python/jittor/src/mem/allocator/cuda_dual_allocator.h @@ -30,7 +30,7 @@ EXTERN_LIB bool no_cuda_error_when_free; struct CudaDualAllocator : Allocator { //for recycle block_id - static const size_t ID_LIMIT = 1 << 16; + static const size_t ID_LIMIT = 1 << 20; int n_free_ids; int free_ids[ID_LIMIT]; DualAllocation allocations[ID_LIMIT]; diff --git a/python/jittor/src/mem/allocator/sfrl_allocator.h b/python/jittor/src/mem/allocator/sfrl_allocator.h index 2bd9a170..3d409e41 100644 --- a/python/jittor/src/mem/allocator/sfrl_allocator.h +++ b/python/jittor/src/mem/allocator/sfrl_allocator.h @@ -36,7 +36,7 @@ struct CachingBlockPool { //start from 1 static size_t tot_block_id; static std::unique_ptr occupied_id_mapper; - static const size_t ID_LIMIT = 1 << 18; + static const size_t ID_LIMIT = 1 << 21; pair get_key(CachingBlock* block); diff --git a/python/jittor/src/mem/allocator/temp_allocator.h b/python/jittor/src/mem/allocator/temp_allocator.h index 08dc0994..ce2cd3b5 100644 --- a/python/jittor/src/mem/allocator/temp_allocator.h +++ b/python/jittor/src/mem/allocator/temp_allocator.h @@ -23,7 +23,7 @@ struct TempCachingBlock { struct TempAllocator : Allocator { static const size_t ALIGN_SIZE = 512; - static const size_t ID_LIMIT = 1 << 18; + static const size_t ID_LIMIT = 1 << 21; static vector temp_allocators; Allocator* underlying; size_t cache_blocks_limit, used_memory, unused_memory;