From 1d9eddff8f84f7750e587cbca1107265db1aaa7c Mon Sep 17 00:00:00 2001 From: nihui Date: Fri, 18 Apr 2025 18:49:01 +0800 Subject: [PATCH] reset extractor local vulkan allocator (#6002) resolve reclaim_blob_allocator get wild allocator warning when clear() called multple times --- src/net.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/net.cpp b/src/net.cpp index 1a130c56c..529da1def 100644 --- a/src/net.cpp +++ b/src/net.cpp @@ -2531,10 +2531,12 @@ void Extractor::clear() if (d->local_blob_vkallocator) { d->net->vulkan_device()->reclaim_blob_allocator(d->local_blob_vkallocator); + d->local_blob_vkallocator = 0; } if (d->local_staging_vkallocator) { d->net->vulkan_device()->reclaim_staging_allocator(d->local_staging_vkallocator); + d->local_staging_vkallocator = 0; } } #endif // NCNN_VULKAN