Add Frontier fidelity envelope campaign
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user