From 7d9d9b83f227c6083007dd78b13ef7826e0efecc Mon Sep 17 00:00:00 2001
From: anurag <anurag@amd.com>
Date: Fri, 11 Sep 2026 11:12:26 -0600
Subject: [PATCH] dump l3 once in skip_iter2

Signed-off-by: anurag <anurag@amd.com>
---
 src/mldebug/batch_runner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mldebug/batch_runner.py b/src/mldebug/batch_runner.py
index 2f68124..c14098b 100644
--- a/src/mldebug/batch_runner.py
+++ b/src/mldebug/batch_runner.py
@@ -422,7 +422,7 @@ def _run_stamp(self, layer, sid, target_itr, cur_it=1):
           self.state.error = True
           break
 
-    if sid == 0:
+    if sid == 0 and not self.args.run_flags.skip_iter2:
       self.dumper.dump_l3_buffers(layer)
     return not self.state.error