gahow/aituner
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Match Frontier to exact iteration composition

Browse Source
This commit is contained in:
Gahow Wang
2026-07-19 19:00:18 +08:00
parent 3349b23290
commit 9c9479c313
2 changed files with 210 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
runs/frontier-fidelity-envelope-v1/test_fidelity_envelope.py
View File

@@ -218,6 +218,21 @@ class FidelityEnvelopeTest(unittest.TestCase):
],
)
def test_qwen235_iteration_state_parser_excludes_warmup(self) -> None:
module = load("analyze_qwen235_fixed_pd_state.py")
rows = [
"Iteration(0): 1 context requests, 512 context tokens, 0 generation requests, 0 generation tokens, iteration elapsed time: 10.00 ms",
"Iteration(1): 1 context requests, 4096 context tokens, 4 generation requests, 4 generation tokens, iteration elapsed time: 30.00 ms",
"Iteration(2): 0 context requests, 0 context tokens, 5 generation requests, 5 generation tokens, iteration elapsed time: 20.00 ms",
]
with tempfile.TemporaryDirectory() as temporary:
path = Path(temporary) / "server.log"
path.write_text("\n".join(rows) + "\n")
parsed = module.parse_iteration_log(path)
self.assertEqual([row["index"] for row in parsed], [1, 2])
self.assertEqual(parsed[0]["context_tokens"], 4096)
self.assertEqual(parsed[1]["generation_requests"], 5)
def test_qwen235_component_categories_are_additive(self) -> None:
module = load("analyze_qwen235_fixed_pd_state.py")
components = {