Function check_alpha_beta

tests/test_device/test_alpha_beta.py:9–15

(rank, world_size, port, physical_devices)

def check_alpha_beta(rank, world_size, port, physical_devices):
    disable_existing_loggers()
    launch(rank=rank, world_size=world_size, host="localhost", port=port, backend="nccl")
    profiler = AlphaBetaProfiler(physical_devices)
    ab_dict = profiler.profile_ab()
    for _, (alpha, beta) in ab_dict.items():
        assert alpha > 0 and alpha < 1e-4 and beta > 0 and beta < 1e-10


@pytest.mark.skip(reason="Skip because assertion fails for CI devices")