From 7231257efc509bfd0c98af1c12a4340610c3ed94 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 29 Dec 2021 17:49:56 +0800 Subject: [PATCH] fix(imperative/fastrun): fix worksapce limit for cpu compnode GitOrigin-RevId: 4583ce6d4b2138b5de0d9de98897b5a3aa73b545 --- src/core/include/megbrain/comp_node.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/core/include/megbrain/comp_node.h b/src/core/include/megbrain/comp_node.h index 56a281f7c..787054a90 100644 --- a/src/core/include/megbrain/comp_node.h +++ b/src/core/include/megbrain/comp_node.h @@ -577,7 +577,7 @@ protected: virtual size_t get_max_reserved_memory() { return 0; } virtual size_t get_max_used_memory() { return 0; } virtual size_t get_max_block_size_available() { return 0; } - virtual size_t get_free_mem() { return 0; } + virtual size_t get_free_mem() { return get_mem_status_bytes().second; } virtual void reset_max_reserved_memory() {} virtual void reset_max_used_memory() {} #endif -- GitLab