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<CachingBlock*[]> occupied_id_mapper;              
-    static const size_t ID_LIMIT = 1 << 18;
+    static const size_t ID_LIMIT = 1 << 21;
 
     pair<size_t,size_t> 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<TempAllocator*> temp_allocators;
     Allocator* underlying;
     size_t cache_blocks_limit, used_memory, unused_memory;