def METHOD_NAME(self):
    with jt.log_capture_scope(enable_tuner=1, log_silent=1,
        log_v=1, log_vprefix="op.cc=100,exe=1000"
    ) as raw_log:
        data = jt.random([5, 5])
        if mpi.world_rank() == 0:
            x = data
        else:
            x = jt.zeros([5, 5])
        y = x.mpi_broadcast(0)
        assert np.allclose(y.data, data.data)
        g = jt.grad(y.sum(),x)
        g_ = g.data
        if mpi.world_rank() == 0:
            assert np.allclose(g_, np.ones([5,5])*n)
    logs = find_log_with_re(raw_log, "(Jit op key (not )?found: nccl_broadcast.*)")
    assert len(logs)==1, len(logs)