add clear device mem pool util (#1153)

* add clear device mem pool util * Apply code-format changes * rename api * Apply code-format changes --------- Co-authored-by: curioyang <curioyang@users.noreply.github.com>
kendryte · Dec 27, 2023 · 52fe13b · 52fe13b
1 parent 907ae20
commit 52fe13b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/src/Native/include/nncase/runtime/allocator.h b/src/Native/include/nncase/runtime/allocator.h
@@ -43,6 +43,7 @@ class NNCASE_API buffer_allocator {
                                     const buffer_attach_options &options) = 0;
 
     static buffer_allocator &host();
+    virtual void shrink_memory_pool() = 0;
 };
 
 END_NS_NNCASE_RUNTIME
diff --git a/src/Native/include/nncase/runtime/util.h b/src/Native/include/nncase/runtime/util.h
@@ -587,4 +587,8 @@ inline dims_t to_4d(dims_t in_a_shape) {
     return in_a_shape;
 }
 
+inline void shrink_memory_pool() {
+    buffer_allocator::host().shrink_memory_pool();
+}
+
 END_NS_NNCASE_RUNTIME
diff --git a/src/Native/src/runtime/host_allocator.cpp b/src/Native/src/runtime/host_allocator.cpp
@@ -122,6 +122,8 @@ class host_buffer_allocator : public buffer_allocator {
             []([[maybe_unused]] gsl::byte *p) {}, paddr, *this,
             host_sync_status_t::valid));
     }
+
+    void shrink_memory_pool() override {}
 };
 
 host_buffer_allocator host_allocator;