model-net-method.h 2.41 KB
Newer Older
1
/*
Philip Carns's avatar
Philip Carns committed
2
 * Copyright (C) 2013 University of Chicago.
3
 * See COPYRIGHT notice in top-level directory.
Philip Carns's avatar
Philip Carns committed
4
 *
5 6 7 8 9 10 11
 */

#ifndef MODELNET_METHOD_H
#define MODELNET_METHOD_H

#include <ross.h>

12 13 14 15 16 17
// 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
18 19
// interface that each model-net model implements

20 21
struct model_net_method
{
22
    uint64_t packet_size; /* packet size */
23
    void (*mn_configure)(); /* For initializing the network */
24 25 26 27 28
    /* 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);
29
    tw_stime (*model_net_method_packet_event)(
30 31 32 33 34 35 36 37 38
            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);
39
    void (*model_net_method_packet_event_rc)(tw_lp *sender);
40 41 42
    tw_stime (*model_net_method_recv_msg_event)(
            const char * category,
            tw_lpid final_dest_lp,
43
            tw_lpid src_mn_lp, // the modelnet LP this message came from
44 45 46 47 48 49 50 51 52
            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);
53 54
    const tw_lptype* (*mn_get_lp_type)();
    int (*mn_get_msg_sz)();
55
    void (*mn_report_stats)();
56
    void (*mn_collective_call)(char const * category, int message_size, int remote_event_size, const void* remote_event, tw_lp* sender);
57
    void (*mn_collective_call_rc)(int message_size, tw_lp* sender);    
58 59
};

60 61
extern struct model_net_method * method_array[];

62 63 64 65 66 67 68 69 70 71
#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
 */