Enhance hook mechanism in dumper (#19073)

This commit is contained in:
fzyzcjy
2026-02-22 16:13:38 +08:00
committed by GitHub
parent fdf80b5031
commit cc63c99f11
3 changed files with 109 additions and 31 deletions

View File

@@ -108,7 +108,7 @@ class TestEndToEnd(CustomTestCase):
dumper.step()
dumper.dump("tensor_b", tensor * 2)
dumper.step()
dump_dirs.append(Path(d) / f"sglang_dump_{dumper._config.partial_name}")
dump_dirs.append(Path(d) / dumper._config.exp_name)
args = Namespace(
baseline_path=str(dump_dirs[0]),