Function test_sampling_distribution_tp2

tests/test_core.py:86–87

()

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


@pytest.mark.skipif(