model-net-method.h 2.69 KB
Newer Older
1
/*
2
 * Copyright (C) 2013 University of Chicago.
3
 * See COPYRIGHT notice in top-level directory.
4
 *
5 6 7 8 9
 */

#ifndef MODELNET_METHOD_H
#define MODELNET_METHOD_H

Jonathan Jenkins's avatar
Jonathan Jenkins committed
10 11 12 13
#ifdef __cplusplus
extern "C" {
#endif

14 15
#include <ross.h>

16 17 18 19 20 21
// forward decl of model_net_method since we currently have a circular include
// (method needs sched def, sched needs method def)
struct model_net_method;

#include "codes/model-net-sched.h"

Jonathan Jenkins's avatar
Jonathan Jenkins committed
22 23
// interface that each model-net model implements

24 25
struct model_net_method
{
26
    uint64_t packet_size; /* packet size */
27
    void (*mn_configure)(); /* For initializing the network */
28 29 30 31 32
    /* Register lp types with ROSS. This may be left as NULL, in which case the
     * type corresponding to name "modelnet_<type>" will be registered
     * automatically. Most networks don't need this (currently, only dragonfly
     * uses it) */
    void (*mn_register)(tw_lptype *base_type);
33
    tw_stime (*model_net_method_packet_event)(
34 35 36 37 38 39 40 41 42
            model_net_request const * req,
            uint64_t message_offset, // offset in the context of the whole message
            uint64_t packet_size, // needed in case message < packet
            tw_stime offset,
            mn_sched_params const * sched_params,
            void const * remote_event,
            void const * self_event,
            tw_lp *sender,
            int is_last_pckt);
43
    void (*model_net_method_packet_event_rc)(tw_lp *sender);
44 45 46
    tw_stime (*model_net_method_recv_msg_event)(
            const char * category,
            tw_lpid final_dest_lp,
47
            tw_lpid src_mn_lp, // the modelnet LP this message came from
48 49 50 51 52 53 54 55 56
            uint64_t msg_size,
            int is_pull,
            uint64_t pull_size,
            tw_stime offset,
            int remote_event_size,
            const void* remote_event,
            tw_lpid src_lp, // original caller of model_net_(pull_)event
            tw_lp *sender); // lp message is being called from (base LP)
    void (*model_net_method_recv_msg_event_rc)(tw_lp *lp);
57 58
    const tw_lptype* (*mn_get_lp_type)();
    int (*mn_get_msg_sz)();
59
    void (*mn_report_stats)();
60
    void (*mn_collective_call)(char const * category, int message_size, int remote_event_size, const void* remote_event, tw_lp* sender);
61 62 63 64 65
    void (*mn_collective_call_rc)(int message_size, tw_lp* sender);
    event_f mn_sample_fn;
    revent_f mn_sample_rc_fn;
    init_f mn_sample_init_fn;
    final_f mn_sample_fini_fn;
66 67
    void (*mn_trace_register)(st_trace_type *base_type);
    const st_trace_type* (*mn_get_trace_type)();
68 69
};

70 71
extern struct model_net_method * method_array[];

Jonathan Jenkins's avatar
Jonathan Jenkins committed
72 73 74 75
#ifdef __cplusplus
}
#endif

76 77 78 79 80 81 82 83 84 85
#endif /* MODELNET_METHOD_H */

/*
 * Local variables:
 *  c-indent-level: 4
 *  c-basic-offset: 4
 * End:
 *
 * vim: ft=c ts=8 sts=4 sw=4 expandtab
 */