Support non orthogonal parallel axes and explicit replication annotation in dump comparator (#19679)

2026-03-02 18:44:33 +08:00
parent a70dd11011
commit 6980416149
40 changed files with 1451 additions and 1012 deletions
@@ -20,7 +20,7 @@ from sglang.srt.debug_utils.comparator.aligner.unsharder.types import (
    ConcatParams,
    UnsharderPlan,
 )
-from sglang.srt.debug_utils.comparator.dims import ParallelAxis, TokenLayout
+from sglang.srt.debug_utils.comparator.dims_spec import ParallelAxis, TokenLayout
 from sglang.srt.debug_utils.comparator.output_types import (
    ErrorLog,
    NonTensorComparisonRecord,