From 4852177c9eccb06685aade6040345044d73d466f Mon Sep 17 00:00:00 2001 From: Hardik Sharma Date: Wed, 4 Feb 2026 14:03:05 -0800 Subject: [PATCH] Add log level for memory planning info prints. (#17172) Summary: Adds a log level arg to control prints. Differential Revision: D92185559 --- backends/cadence/aot/memory_planning.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backends/cadence/aot/memory_planning.py b/backends/cadence/aot/memory_planning.py index 5ff39580a23..43a96ee690d 100644 --- a/backends/cadence/aot/memory_planning.py +++ b/backends/cadence/aot/memory_planning.py @@ -290,6 +290,7 @@ def print_memory_planning_info( opt_level: int, alloc_graph_input: bool, alloc_graph_output: bool, + log_level = logging.INFO, ) -> None: # Get the peak memory usages per memory space mem_constraints = MemConstraints( @@ -319,7 +320,8 @@ def print_memory_planning_info( ] # Print the memory usage per memory space as a table - logging.info( + logging.log( + log_level, "\n" + tabulate( memory_usage_table, @@ -352,7 +354,8 @@ def print_memory_planning_info( ] # Print the total memory usage as a table - logging.info( + logging.log( + log_level, "\n" + tabulate( total_memory_usage_table,