diff --git a/src/mldebug/backend/test_impl.py b/src/mldebug/backend/test_impl.py index 4949bf1..4e17033 100644 --- a/src/mldebug/backend/test_impl.py +++ b/src/mldebug/backend/test_impl.py @@ -316,7 +316,7 @@ def print_pc_breakpoints(self): Returns: None """ - print(f"Currently configured PC Breakpoints: {self.pc_brkpts}") + print(f"{self.pc_brkpts}") def write_aie_regs(self, reg, value): """ diff --git a/src/mldebug/backend/xrt_impl.py b/src/mldebug/backend/xrt_impl.py index e5643cf..de414ea 100644 --- a/src/mldebug/backend/xrt_impl.py +++ b/src/mldebug/backend/xrt_impl.py @@ -191,7 +191,7 @@ def print_pc_breakpoints(self): Returns: None """ - print(f"Currently configured PC Breakpoints: {self.pc_brkpts}") + print(f"{self.pc_brkpts}") def enable_pc_halt(self): """ diff --git a/src/mldebug/client_debug.py b/src/mldebug/client_debug.py index ed79832..7173621 100644 --- a/src/mldebug/client_debug.py +++ b/src/mldebug/client_debug.py @@ -355,9 +355,11 @@ def rreg_stamp(self, offset, sid=0): def print_core_summary(self): """Print the core summary for all stamps.""" self.status_handle.print_core_summary() + print("===== Configured PC Breakpoints =====") for sid, impl in enumerate(self.impls): - print(f"\n=== Stamp {sid} PC Breakpoints ===") + print(f"[Stamp_{sid}]: ", end="") impl.print_pc_breakpoints() + print() self.print_current_state() print("[INFO] Currently only leftmost stamp is supported for advanced debug.") diff --git a/src/mldebug/input_parser.py b/src/mldebug/input_parser.py index ba5cb59..59e6eba 100644 --- a/src/mldebug/input_parser.py +++ b/src/mldebug/input_parser.py @@ -145,6 +145,10 @@ def get_flag(s, default=False): get_flag("disable_tg"), ) + # Support all stamps in standalone mode + if args.aie_only: + args.run_flags.multistamp=True + def set_aie_scc(args, subgraph_path): """ diff --git a/src/mldebug/mldebug_cli.py b/src/mldebug/mldebug_cli.py index 08cc501..c6afb8e 100644 --- a/src/mldebug/mldebug_cli.py +++ b/src/mldebug/mldebug_cli.py @@ -79,7 +79,7 @@ def check_args(args): args.interactive = False args.aie_only = True print("[INFO] Dumping advanced AIE status and exiting (non-interactive)") - elif args.exec_cmd is not None: + elif args.exec_cmd: args.aie_only = True args.interactive = False elif args.aie_only: @@ -94,12 +94,16 @@ def check_args(args): if not args.dump_aie_status: args.interactive = True print("[INFO] Using standalone mode for core dumps") + if args.backend == "core_dump" and args.core_dump is None: print( "[ERROR] Core dump file is required when backend is 'core_dump'. Please use -h or --help for usage" ) + + # Default overlay for this device if args.device == AIE_DEV_NPU3 and not args.overlay: args.overlay = "3x4" + return True