Add Frontier fidelity envelope campaign

This commit is contained in:
2026-07-17 09:44:49 +08:00
parent 3a59d5df96
commit 95f4af3d99
17 changed files with 688 additions and 1 deletions

View File

@@ -43,3 +43,20 @@ def test_kendall_tau_b() -> None:
analysis = load("analyze_qwen30_prefill_fidelity.py")
assert analysis.kendall_tau_b([1, 2, 3], [1, 2, 3])["kendall_tau_b"] == 1
assert analysis.kendall_tau_b([1, 2, 3], [3, 2, 1])["kendall_tau_b"] == -1
def test_configure_cc_command(tmp_path: Path) -> None:
surface = load("run_frontier_qwen30_prefill_surface.py")
base = ["python", "--cc_backend_config_type", "analytical", "--other", "x"]
analytical = surface.configure_cc_command(
base, backend="analytical", allreduce_csv=None, cache=tmp_path
)
assert analytical == base
profile = tmp_path / "all_reduce.csv"
profile.write_text("header\n")
vidur = surface.configure_cc_command(
base, backend="vidur", allreduce_csv=profile, cache=tmp_path / "cache"
)
assert vidur[2] == "vidur"
assert "--vidur_cc_backend_config_all_reduce_input_file" in vidur
assert str(profile) in vidur