From cb2dd1a10614c21ca78e1c0cea2f6f7aa882e712 Mon Sep 17 00:00:00 2001 From: oahzxl Date: Tue, 27 Dec 2022 15:01:58 +0800 Subject: [PATCH] turn off print mem --- chunk_codegen.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/chunk_codegen.py b/chunk_codegen.py index 3cd10350e..6caed88d8 100644 --- a/chunk_codegen.py +++ b/chunk_codegen.py @@ -1750,10 +1750,13 @@ class ChunkRegionSearch(object): _, active_node, ) = self.memory_estimator.estimate_chunk_inference_mem( - self.index_tracer.node_list, chunk_infos, print_mem=True + self.index_tracer.node_list, chunk_infos ) if self._stop_search(init_mem_peak, mem_peak): break + # self.memory_estimator.estimate_chunk_inference_mem( + # self.index_tracer.node_list, chunk_infos, print_mem=True + # ) return chunk_infos