diff --git a/src/core/include/megbrain/comp_node.h b/src/core/include/megbrain/comp_node.h index 56a281f7..787054a9 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