Function test_greedy_tp2

tests/test_core.py:98–99

()

    reason="Requires >= 2 GPUs (symmetric memory needs NVLink)",
)
def test_greedy_tp2() -> None:
    run_maybe_distributed(verify_greedy_tp2, n_procs=2)


@pytest.mark.parametrize("n_hidden_states", [1, 2])