diag: add per-segment debug output for D5c multi-tile

2026-05-26 15:28:19 +00:00
parent e64392f1ac
commit bbc29945e8
1 changed files with 3 additions and 0 deletions
--- a/tests/unit/test_d5c_multitile.py
+++ b/tests/unit/test_d5c_multitile.py
@@ -247,6 +247,9 @@ def test_d5c_multitile():
    if cos < 0.99:
        print(f'  kernel[0,:4]={o_merged[0,:4].tolist()}')
        print(f'  ref[0,:4]={ref[0,:4].tolist()}')
+        # Debug: check each segment individually
+        for i, (o_n, lse) in enumerate(segment_results):
+            print(f'  Seg {i}: LSE[0]={lse[0].item():.4f}  O_norm[0,:2]={o_n[0,:2].tolist()}')


 if __name__ == '__main__':