File: btl_portals4_frag.h

package info (click to toggle)
openmpi 5.0.7-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, trixie
  • size: 202,312 kB
  • sloc: ansic: 612,441; makefile: 42,495; sh: 11,230; javascript: 9,244; f90: 7,052; java: 6,404; perl: 5,154; python: 1,856; lex: 740; fortran: 61; cpp: 20; tcl: 12
file content (136 lines) | stat: -rw-r--r-- 5,680 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
/*
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2006 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  All rights
 *                         reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * Copyright (c) 2011-2015 Los Alamos National Security, LLC.
 *                         All rights reserved.
 * Copyright (c) 2014      Bull SAS.  All rights reserved.
 * $COPYRIGHT$
 *
 * Additional copyrights may follow
 *
 * $HEADER$
 */

#ifndef OPAL_BTL_PORTALS4_FRAG_H
#define OPAL_BTL_PORTALS4_FRAG_H

#include "opal_config.h"

#include "btl_portals4.h"

BEGIN_C_DECLS

struct mca_btl_portals4_segment_t {
    mca_btl_base_segment_t base;
    ptl_match_bits_t key;
};
typedef struct mca_btl_portals4_segment_t mca_btl_portals4_segment_t;

/**
 * Portals send fragment derived type
 */
struct mca_btl_portals4_frag_t {
    mca_btl_base_descriptor_t base;
    mca_btl_portals4_segment_t segments[1];
    /* needed for retransmit case */
    struct mca_btl_base_endpoint_t *endpoint;
    /* needed for retransmit case */
    mca_btl_base_header_t hdr;
    /* handle to use for communication */
    ptl_handle_me_t me_h;
    /* size of the allocated memory region -- not the amount of data
       we need to send */
    void *addr;
    size_t size;
    /* match bits for retransmit case */
    ptl_match_bits_t match_bits;
    /* length for retransmit case */
    ptl_size_t length;
    /* length for retransmit case */
    ptl_process_t peer_proc;

    /* the callback and context to complete an RDMA operation */
    struct {
        mca_btl_base_rdma_completion_fn_t func;
        void *context;
        void *data;
        mca_btl_base_registration_handle_t *local_handle;
    } rdma_cb;

    enum {
        BTL_PORTALS4_FRAG_TYPE_EAGER,
        BTL_PORTALS4_FRAG_TYPE_MAX,
        BTL_PORTALS4_FRAG_TYPE_USER
    } type;
    unsigned char data[16];
};
typedef struct mca_btl_portals4_frag_t mca_btl_portals4_frag_t;
OBJ_CLASS_DECLARATION(mca_btl_portals4_frag_t);

typedef struct mca_btl_portals4_frag_t mca_btl_portals4_frag_eager_t;
OBJ_CLASS_DECLARATION(mca_btl_portals4_frag_eager_t);

typedef struct mca_btl_portals4_frag_t mca_btl_portals4_frag_max_t;
OBJ_CLASS_DECLARATION(mca_btl_portals4_frag_max_t);

typedef struct mca_btl_portals4_frag_t mca_btl_portals4_frag_user_t;
OBJ_CLASS_DECLARATION(mca_btl_portals4_frag_user_t);

/*
 * Macros to allocate/return descriptors from module specific
 * free list(s).
 */
#define OPAL_BTL_PORTALS4_FRAG_ALLOC_EAGER(btl_macro, frag)                  \
    {                                                                        \
        frag = (mca_btl_portals4_frag_t *) opal_free_list_get(               \
            &((mca_btl_portals4_module_t *) btl_macro)->portals_frag_eager); \
        if (NULL == frag) {                                                  \
            OPAL_BTL_PORTALS4_FRAG_ALLOC_MAX(btl_macro, frag);               \
        }                                                                    \
    }

#define OPAL_BTL_PORTALS4_FRAG_RETURN_EAGER(btl_macro, frag)                                  \
    {                                                                                         \
        assert(BTL_PORTALS4_FRAG_TYPE_EAGER == frag->type);                                   \
        opal_free_list_return(&((mca_btl_portals4_module_t *) btl_macro)->portals_frag_eager, \
                              (opal_free_list_item_t *) (frag));                              \
    }

#define OPAL_BTL_PORTALS4_FRAG_ALLOC_MAX(btl_macro, frag)                  \
    {                                                                      \
        frag = (mca_btl_portals4_frag_t *) opal_free_list_get(             \
            &((mca_btl_portals4_module_t *) btl_macro)->portals_frag_max); \
    }

#define OPAL_BTL_PORTALS4_FRAG_RETURN_MAX(btl_macro, frag)                                  \
    {                                                                                       \
        assert(BTL_PORTALS4_FRAG_TYPE_MAX == frag->type);                                   \
        opal_free_list_return(&((mca_btl_portals4_module_t *) btl_macro)->portals_frag_max, \
                              (opal_free_list_item_t *) (frag));                            \
    }

#define OPAL_BTL_PORTALS4_FRAG_ALLOC_USER(btl_macro, frag)                  \
    {                                                                       \
        frag = (mca_btl_portals4_frag_t *) opal_free_list_get(              \
            &((mca_btl_portals4_module_t *) btl_macro)->portals_frag_user); \
        frag->base.des_cbfunc = NULL;                                       \
    }

#define OPAL_BTL_PORTALS4_FRAG_RETURN_USER(btl_macro, frag)                                  \
    {                                                                                        \
        assert(BTL_PORTALS4_FRAG_TYPE_USER == frag->type);                                   \
        opal_free_list_return(&((mca_btl_portals4_module_t *) btl_macro)->portals_frag_user, \
                              (opal_free_list_item_t *) (frag));                             \
    }

END_C_DECLS
#endif