From 53a02dd7f9a98f1ec481a9c6b00b5263d0261bd2 Mon Sep 17 00:00:00 2001 From: Mohammad Angkad Date: Fri, 2 Oct 2026 03:10:11 +0000 Subject: [PATCH] Fix stale source-patch match in dumper comparator e2e test --- .../debug_utils/test_engine_dumper_comparator_e2e.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/registered/debug_utils/test_engine_dumper_comparator_e2e.py b/test/registered/debug_utils/test_engine_dumper_comparator_e2e.py index 510a0d678228..6176ecc70e21 100644 --- a/test/registered/debug_utils/test_engine_dumper_comparator_e2e.py +++ b/test/registered/debug_utils/test_engine_dumper_comparator_e2e.py @@ -71,8 +71,8 @@ hidden_states = self.attn_boundary.prepare( hidden_states, forward_batch, - captured_last_layer_outputs=captured_last_layer_outputs, - capture_output=capture_output, + capture_gathered=captured_last_layer_outputs, + capture=capture_output, **kwargs, ) append: "dumper.dump('layer_input', hidden_states, dims='t h # tp:replicated')" @@ -123,8 +123,8 @@ hidden_states = self.attn_boundary.prepare( hidden_states, forward_batch, - captured_last_layer_outputs=captured_last_layer_outputs, - capture_output=capture_output, + capture_gathered=captured_last_layer_outputs, + capture=capture_output, **kwargs, ) append: "dumper.dump('layer_input', hidden_states, dims='t h # tp:replicated dp:=attn_dp')"