Enhance displaying and debuggability in dump comparator (#19466)

This commit is contained in:
fzyzcjy
2026-02-27 09:06:19 +08:00
committed by GitHub
parent 23cbbd6d41
commit 4add6ec0f6
17 changed files with 879 additions and 31 deletions
@@ -124,7 +124,6 @@ class TestExecuteAlignerPlan:
y=[self._make_step_plan(step=0, indices=[0])],
),
token_aligner_plan=None,
token_dims=Pair(x=0, y=0),
)
tensors_pair: Pair[list[torch.Tensor]] = Pair(
@@ -146,7 +145,6 @@ class TestExecuteAlignerPlan:
y=[self._make_step_plan(step=0, indices=[0, 1])],
),
token_aligner_plan=None,
token_dims=Pair(x=0, y=0),
)
tensors_pair: Pair[list[torch.Tensor]] = Pair(
@@ -168,7 +166,6 @@ class TestExecuteAlignerPlan:
y=[self._make_step_plan(step=0, indices=[0])],
),
token_aligner_plan=None,
token_dims=Pair(x=0, y=0),
)
t_x: torch.Tensor = torch.tensor([1.0, 2.0])
@@ -191,7 +188,6 @@ class TestExecuteAlignerPlan:
y=[self._make_step_plan(step=0, indices=[0])],
),
token_aligner_plan=None,
token_dims=Pair(x=0, y=0),
)
tensors_pair: Pair[list[torch.Tensor]] = Pair(