turn off print mem

cb2dd1a1 · oahzxl · a2b4755c · cb2dd1a1
Commit cb2dd1a1 authored Dec 27, 2022 by oahzxl
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

chunk_codegen.py chunk_codegen.py +4 -1

No files found.
--- 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