test_exchange.c File Reference

#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include <math.h>
#include <mpi.h>
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <gsl/gsl_rng.h>
#include <libgadget/exchange.h>
#include <libgadget/domain.h>
#include <libgadget/slotsmanager.h>
#include <libgadget/partmanager.h>
#include "stub.h"
#include <libgadget/walltime.h>

Include dependency graph for test_exchange.c:

Go to the source code of this file.

Macros
#define	qsort_openmp   qsort
#define	NUMPART1   8

Functions
static int	setup_particles (int64_t NType[6])
static int	teardown_particles (void **state)
static int	test_exchange_layout_func (int i, const void *userdata)
static void	test_exchange (void **state)
static void	test_exchange_zero_slots (void **state)
static void	test_exchange_with_garbage (void **state)
static int	test_exchange_layout_func_uneven (int i, const void *userdata)
static void	test_exchange_uneven (void **state)
int	main (void)

Variables
int	NTask
int	ThisTask
int	TotNumPart
static struct ClockTable	Clocks

Macro Definition Documentation

NUMPART1

#define NUMPART1   8

Definition at line 28 of file test_exchange.c.

qsort_openmp

#define qsort_openmp   qsort

Definition at line 14 of file test_exchange.c.

Function Documentation

main()

int main

(

void

)

Definition at line 210 of file test_exchange.c.

               {
    const struct CMUnitTest tests[] = {
        cmocka_unit_test(test_exchange_with_garbage),
        cmocka_unit_test(test_exchange),
        cmocka_unit_test(test_exchange_zero_slots),
        cmocka_unit_test(test_exchange_uneven),
    };
    return cmocka_run_group_tests_mpi(tests, NULL, NULL);
}

References test_exchange(), test_exchange_uneven(), test_exchange_with_garbage(), and test_exchange_zero_slots().

Here is the call graph for this function:

setup_particles()

static int setup_particles ( int64_t  NType[6] )

static

Definition at line 30 of file test_exchange.c.

{
    walltime_init(&Clocks);
    MPI_Barrier(MPI_COMM_WORLD);
    PartManager->MaxPart = 1024;
    int ptype;
    PartManager->NumPart = 0;
    for(ptype = 0; ptype < 6; ptype ++) {
        PartManager->NumPart += NType[ptype];
    }
    MPI_Comm_rank(MPI_COMM_WORLD, &ThisTask);
    MPI_Comm_size(MPI_COMM_WORLD, &NTask);
 
    P = (struct particle_data *) mymalloc("P", PartManager->MaxPart * sizeof(struct particle_data));
    memset(P, 0, sizeof(struct particle_data) * PartManager->MaxPart);
 
    slots_init(0.01 * PartManager->MaxPart, SlotsManager);
    slots_set_enabled(0, sizeof(struct sph_particle_data), SlotsManager);
    slots_set_enabled(4, sizeof(struct star_particle_data), SlotsManager);
    slots_set_enabled(5, sizeof(struct bh_particle_data), SlotsManager);
 
 
    slots_reserve(1, NType, SlotsManager);
 
    slots_setup_topology(PartManager, NType, SlotsManager);
 
    int i;
    #pragma omp parallel for
    for(i = 0; i < PartManager->NumPart; i ++) {
        P[i].ID = i + PartManager->NumPart * ThisTask;
    }
 
    slots_setup_id(PartManager, SlotsManager);
 
    MPI_Allreduce(&PartManager->NumPart, &TotNumPart, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
 
    return 0;
}

References Clocks, part_manager_type::MaxPart, mymalloc, NTask, part_manager_type::NumPart, P, PartManager, ptype, slots_init(), slots_reserve(), slots_set_enabled(), slots_setup_id(), slots_setup_topology(), SlotsManager, ThisTask, TotNumPart, and walltime_init().

Referenced by test_exchange(), test_exchange_uneven(), test_exchange_with_garbage(), and test_exchange_zero_slots().

Here is the call graph for this function:

Here is the caller graph for this function:

teardown_particles()

static int teardown_particles ( void **  state )

static

Definition at line 70 of file test_exchange.c.

{
    int TotNumPart2;
 
    MPI_Allreduce(&PartManager->NumPart, &TotNumPart2, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD);
 
    assert_int_equal(TotNumPart2, TotNumPart);
 
    slots_free(SlotsManager);
    myfree(P);
    MPI_Barrier(MPI_COMM_WORLD);
    return 0;
}

References myfree, part_manager_type::NumPart, P, PartManager, slots_free(), SlotsManager, and TotNumPart.

Referenced by test_exchange(), test_exchange_uneven(), test_exchange_with_garbage(), and test_exchange_zero_slots().

Here is the call graph for this function:

Here is the caller graph for this function:

test_exchange()

static void test_exchange ( void **  state )

static

Definition at line 92 of file test_exchange.c.

{
    int64_t newSlots[6] = {NUMPART1, NUMPART1, NUMPART1, NUMPART1, NUMPART1, NUMPART1};
 
    setup_particles(newSlots);
 
    int i;
 
    int fail = domain_exchange(&test_exchange_layout_func, NULL, 1, NULL, PartManager, SlotsManager,10000, MPI_COMM_WORLD);
 
    assert_all_true(!fail);
 
    slots_check_id_consistency(PartManager, SlotsManager);
    domain_test_id_uniqueness(PartManager);
 
    for(i = 0; i < PartManager->NumPart; i ++) {
        assert_true(P[i].ID % NTask == 1Lu * ThisTask);
    }
 
    teardown_particles(state);
    return;
}

References domain_exchange(), domain_test_id_uniqueness(), particle_data::ID, NTask, part_manager_type::NumPart, NUMPART1, P, PartManager, setup_particles(), slots_check_id_consistency(), SlotsManager, teardown_particles(), test_exchange_layout_func(), and ThisTask.

Referenced by main().

Here is the call graph for this function:

Here is the caller graph for this function:

test_exchange_layout_func()

static int test_exchange_layout_func ( int  i, const void *  userdata  )

static

Definition at line 86 of file test_exchange.c.

{
    return P[i].ID % NTask;
}

References NTask, and P.

Referenced by test_exchange(), test_exchange_with_garbage(), and test_exchange_zero_slots().

Here is the caller graph for this function:

test_exchange_layout_func_uneven()

static int test_exchange_layout_func_uneven ( int  i, const void *  userdata  )

static

Definition at line 170 of file test_exchange.c.

{
    if(P[i].Type == 0) return 0;
 
    return P[i].ID % NTask;
}

References NTask, P, and particle_data::Type.

Referenced by test_exchange_uneven().

Here is the caller graph for this function:

test_exchange_uneven()

static void test_exchange_uneven ( void **  state )

static

Definition at line 178 of file test_exchange.c.

{
    int64_t newSlots[6] = {NUMPART1, NUMPART1, NUMPART1, NUMPART1, NUMPART1, NUMPART1};
 
    setup_particles(newSlots);
    int i;
 
    /* this will trigger a slot growth on slot type 0 due to the inbalance */
    int fail = domain_exchange(&test_exchange_layout_func_uneven, NULL, 1, NULL, PartManager, SlotsManager, 10000, MPI_COMM_WORLD);
 
    assert_all_true(!fail);
 
    if(ThisTask == 0) {
        /* the slot type must have grown automatically to handle the new particles. */
        assert_int_equal(SlotsManager->info[0].size, NUMPART1 * NTask);
    }
 
    slots_check_id_consistency(PartManager, SlotsManager);
    domain_test_id_uniqueness(PartManager);
 
    for(i = 0; i < PartManager->NumPart; i ++) {
        if(P[i].Type == 0) {
            assert_true (ThisTask == 0);
        } else {
            assert_true(P[i].ID % NTask == 1Lu * ThisTask);
        }
    }
 
    teardown_particles(state);
    return;
}

References domain_exchange(), domain_test_id_uniqueness(), particle_data::ID, slots_manager_type::info, NTask, part_manager_type::NumPart, NUMPART1, P, PartManager, setup_particles(), slot_info::size, slots_check_id_consistency(), SlotsManager, teardown_particles(), test_exchange_layout_func_uneven(), ThisTask, and particle_data::Type.

Referenced by main().

Here is the call graph for this function:

Here is the caller graph for this function:

test_exchange_with_garbage()

static void test_exchange_with_garbage ( void **  state )

static

Definition at line 140 of file test_exchange.c.

{
    int64_t newSlots[6] = {NUMPART1, NUMPART1, NUMPART1, NUMPART1, NUMPART1, NUMPART1};
 
    setup_particles(newSlots);
    int i;
 
    slots_mark_garbage(0, PartManager, SlotsManager); /* watch out! this propagates the garbage flag to children */
    TotNumPart -= NTask;
 
    int fail = domain_exchange(&test_exchange_layout_func, NULL, 1, NULL, PartManager, SlotsManager, 10000, MPI_COMM_WORLD);
 
    assert_all_true(!fail);
 
    domain_test_id_uniqueness(PartManager);
    slots_check_id_consistency(PartManager, SlotsManager);
 
    for(i = 0; i < PartManager->NumPart; i ++) {
        assert_true (P[i].ID % NTask == 1Lu * ThisTask);
    }
 
    for(i = 0; i < PartManager->NumPart; i ++) {
        assert_true (P[i].IsGarbage == 0);
    }
 
    teardown_particles(state);
    return;
}

References domain_exchange(), domain_test_id_uniqueness(), particle_data::ID, particle_data::IsGarbage, NTask, part_manager_type::NumPart, NUMPART1, P, PartManager, setup_particles(), slots_check_id_consistency(), slots_mark_garbage(), SlotsManager, teardown_particles(), test_exchange_layout_func(), ThisTask, and TotNumPart.

Referenced by main().

Here is the call graph for this function:

Here is the caller graph for this function:

test_exchange_zero_slots()

static void test_exchange_zero_slots ( void **  state )

static

Definition at line 116 of file test_exchange.c.

{
    int64_t newSlots[6] = {NUMPART1, 0, NUMPART1, 0, NUMPART1, 0};
 
    setup_particles(newSlots);
 
    int i;
 
    int fail = domain_exchange(&test_exchange_layout_func, NULL, 1, NULL, PartManager, SlotsManager, 10000, MPI_COMM_WORLD);
 
    assert_all_true(!fail);
 
    slots_check_id_consistency(PartManager, SlotsManager);
    domain_test_id_uniqueness(PartManager);
 
    for(i = 0; i < PartManager->NumPart; i ++) {
        assert_true (P[i].ID % NTask == 1Lu*ThisTask);
    }
 
    teardown_particles(state);
    return;
}

References domain_exchange(), domain_test_id_uniqueness(), particle_data::ID, NTask, part_manager_type::NumPart, NUMPART1, P, PartManager, setup_particles(), slots_check_id_consistency(), SlotsManager, teardown_particles(), test_exchange_layout_func(), and ThisTask.

Referenced by main().

Here is the call graph for this function:

Here is the caller graph for this function:

Variable Documentation

Clocks

struct ClockTable Clocks

static

Definition at line 24 of file test_exchange.c.

Referenced by setup_particles().

NTask

int NTask

Definition at line 23 of file test_exchange.c.

Referenced by _assign_colors(), _collect_sizes(), _create_segment_group(), _MPIU_Barrier(), _solve_for_layout_mpi(), _transpose_plan_entries(), check_sorted(), cluster_get_num_hosts(), compute_global_quantities_of_system(), count_QSO_halos(), do_long_radix_test(), do_mpsort_test(), domain_allocate(), domain_assign_balanced(), domain_attempt_decompose(), domain_check_memory_bound(), domain_find_iter_space(), domain_nonrecursively_combine_topTree(), domain_policies_init(), domain_test_id_uniqueness(), ev_alloc_threadlocals(), fof_assign_grnr(), fof_reduce_groups(), fof_seed(), fof_try_particle_exchange(), force_exchange_pseudodata(), force_update_hmax(), hci_query_filesystem(), init_alloc_particle_slot_memory(), layout_exchange_pencils(), layout_prepare(), main(), MPI_Alltoallv_smart(), MPI_Alltoallv_sparse(), MPIU_cumsum(), MPIU_Gather(), MPIU_GetLoc(), MPIU_Scatter(), MPIU_write_pids(), mpsort_mpi_newarray_impl(), mymalloc_init(), petapm_init(), report_detailed_memory_usage(), run_gravity_test(), setup_particles(), sumup_large_ints(), sumup_longs(), tamalloc_init(), test_exchange(), test_exchange_layout_func(), test_exchange_layout_func_uneven(), test_exchange_uneven(), test_exchange_with_garbage(), test_exchange_zero_slots(), test_fof(), and walltime_summary_clocks().

ThisTask

int ThisTask

Definition at line 23 of file test_exchange.c.

Referenced by _assign_colors(), _collect_sizes(), _create_segment_group(), _MPIU_Barrier(), _solve_for_layout_mpi(), add_particle_to_group(), check_sorted(), cluster_get_num_hosts(), convert_node_to_region(), count_QSO_halos(), do_long_radix_test(), do_mpsort_test(), domain_build_exchange_list(), domain_determine_global_toptree(), domain_nonrecursively_combine_topTree(), domain_test_id_uniqueness(), endrun(), find_last_snapnum(), fof_assign_grnr(), fof_compile_catalogue(), fof_label_primary(), fof_reduce_groups(), fof_seed(), fof_seed_make_one(), fof_try_particle_exchange(), force_exchange_pseudodata(), force_insert_pseudo_particles(), force_tree_create_nodes(), force_update_hmax(), force_update_node_parallel(), hci_query_filesystem(), init_alloc_particle_slot_memory(), init_powerspectrum(), init_transfer_table(), lightcone_init(), load_heii_reion_hist(), load_treecool(), main(), MPI_Alltoallv_smart(), MPI_Alltoallv_sparse(), MPIU_cumsum(), MPIU_Gather(), MPIU_GetLoc(), MPIU_Scatter(), MPIU_Tracev(), MPIU_write_pids(), mpsort_mpi_newarray_impl(), open_outputfiles(), petaio_read_icnutransfer(), petaio_read_neutrinos(), petaio_read_snapshot(), petaio_save_neutrinos(), petaio_save_snapshot(), petapm_init(), powerspectrum_nu_save(), powerspectrum_save(), print_spec(), read_big_array(), read_parameter_file(), read_parameterfile(), read_power_table(), report_detailed_memory_usage(), save_all_transfer_tables(), save_transfer(), set_all_global_params(), set_blackhole_params(), set_cooling_params(), set_density_params(), set_domain_params(), set_fof_params(), set_gravshort_tree_params(), set_hydro_params(), set_init_params(), set_metal_return_params(), set_petaio_params(), set_qso_lightup_params(), set_sfr_params(), set_stats_params(), set_timestep_params(), set_treewalk_params(), set_winds_params(), setup_glass(), setup_particles(), sumup_large_ints(), sumup_longs(), test_exchange(), test_exchange_uneven(), test_exchange_with_garbage(), test_exchange_zero_slots(), test_fof(), and write_checkpoint().

TotNumPart

int TotNumPart

Definition at line 24 of file test_exchange.c.

Referenced by _bigfile_utils_create_block_from_c_array(), init_alloc_particle_slot_memory(), main(), print_timebin_statistics(), setup_particles(), teardown_particles(), and test_exchange_with_garbage().