"configs/datasets/CLUE_afqmc/CLUE_afqmc_gen_db509b.py" did not exist on "7d346000bb8f1f7611f88dc8e003bdf8c9ae3ece"
corundum_bm.h 7.23 KB
Newer Older
Jialin Li's avatar
Jialin Li committed
1
2
3
4
#pragma once

#include <stdint.h>

Jialin Li's avatar
Jialin Li committed
5
6
typedef uint32_t reg_t;
typedef uint64_t addr_t;
7
typedef uint16_t ptr_t;
Jialin Li's avatar
Jialin Li committed
8

Jialin Li's avatar
Jialin Li committed
9
10
11
12
13
14
15
16
17
18
19
#define REG_FW_ID                   0x0000
#define REG_FW_VER                  0x0004
#define REG_BOARD_ID                0x0008
#define REG_BOARD_VER               0x000C
#define REG_PHC_COUNT               0x0010
#define REG_PHC_OFFSET              0x0014
#define REG_PHC_STRIDE              0x0018
#define REG_IF_COUNT                0x0020
#define REG_IF_STRIDE               0x0024
#define REG_IF_CSR_OFFSET           0x002C

Jialin Li's avatar
Jialin Li committed
20
21
22
23
24
25
#define IF_FEATURE_RSS              (1 << 0)
#define IF_FEATURE_PTP_TS           (1 << 4)
#define IF_FEATURE_TX_CSUM          (1 << 8)
#define IF_FEATURE_RX_CSUM          (1 << 9)
#define IF_FEATURE_RX_HASH          (1 << 10)

Jialin Li's avatar
Jialin Li committed
26
#define PHC_REG_FEATURES            0x0200
Jialin Li's avatar
Jialin Li committed
27
28
#define PHC_REG_PTP_CUR_SEC_L       0x0218
#define PHC_REG_PTP_CUR_SEC_H       0x021C
Jialin Li's avatar
Jialin Li committed
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#define PHC_REG_PTP_SET_FNS         0x0230
#define PHC_REG_PTP_SET_NS          0x0234
#define PHC_REG_PTP_SET_SEC_L       0x0238
#define PHC_REG_PTP_SET_SEC_H       0x023C

#define IF_REG_IF_ID                0x80000
#define IF_REG_IF_FEATURES          0x80004
#define IF_REG_EVENT_QUEUE_COUNT    0x80010
#define IF_REG_EVENT_QUEUE_OFFSET   0x80014
#define IF_REG_TX_QUEUE_COUNT       0x80020
#define IF_REG_TX_QUEUE_OFFSET      0x80024
#define IF_REG_TX_CPL_QUEUE_COUNT   0x80028
#define IF_REG_TX_CPL_QUEUE_OFFSET  0x8002C
#define IF_REG_RX_QUEUE_COUNT       0x80030
#define IF_REG_RX_QUEUE_OFFSET      0x80034
#define IF_REG_RX_CPL_QUEUE_COUNT   0x80038
#define IF_REG_RX_CPL_QUEUE_OFFSET  0x8003C
#define IF_REG_PORT_COUNT           0x80040
#define IF_REG_PORT_OFFSET          0x80044
#define IF_REG_PORT_STRIDE          0x80048

Jialin Li's avatar
Jialin Li committed
50
#define QUEUE_ACTIVE_MASK 0x80000000
51
52
#define QUEUE_ARM_MASK 0x80000000
#define QUEUE_CONT_MASK 0x40000000
Jialin Li's avatar
Jialin Li committed
53

Jialin Li's avatar
Jialin Li committed
54
55
56
57
58
59
60
61
62
63
64
65
#define EVENT_QUEUE_BASE_ADDR_REG       0x100000
#define EVENT_QUEUE_ACTIVE_LOG_SIZE_REG 0x100008
#define EVENT_QUEUE_INTERRUPT_INDEX_REG 0x10000C
#define EVENT_QUEUE_HEAD_PTR_REG        0x100010
#define EVENT_QUEUE_TAIL_PTR_REG        0x100018

#define TX_QUEUE_BASE_ADDR_REG       0x200000
#define TX_QUEUE_ACTIVE_LOG_SIZE_REG 0x200008
#define TX_QUEUE_CPL_QUEUE_INDEX_REG 0x20000C
#define TX_QUEUE_HEAD_PTR_REG        0x200010
#define TX_QUEUE_TAIL_PTR_REG        0x200018

Jialin Li's avatar
Jialin Li committed
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
#define TX_CPL_QUEUE_BASE_ADDR_REG       0x400000
#define TX_CPL_QUEUE_ACTIVE_LOG_SIZE_REG 0x400008
#define TX_CPL_QUEUE_INTERRUPT_INDEX_REG 0x40000C
#define TX_CPL_QUEUE_HEAD_PTR_REG        0x400010
#define TX_CPL_QUEUE_TAIL_PTR_REG        0x400018

#define RX_QUEUE_BASE_ADDR_REG       0x600000
#define RX_QUEUE_ACTIVE_LOG_SIZE_REG 0x600008
#define RX_QUEUE_CPL_QUEUE_INDEX_REG 0x60000C
#define RX_QUEUE_HEAD_PTR_REG        0x600010
#define RX_QUEUE_TAIL_PTR_REG        0x600018

#define RX_CPL_QUEUE_BASE_ADDR_REG       0x700000
#define RX_CPL_QUEUE_ACTIVE_LOG_SIZE_REG 0x700008
#define RX_CPL_QUEUE_INTERRUPT_INDEX_REG 0x70000C
#define RX_CPL_QUEUE_HEAD_PTR_REG        0x700010
#define RX_CPL_QUEUE_TAIL_PTR_REG        0x700018

#define PORT_REG_PORT_ID                    0x800000
#define PORT_REG_PORT_FEATURES              0x800004
#define PORT_REG_PORT_MTU                   0x800008
#define PORT_REG_SCHED_COUNT                0x800010
#define PORT_REG_SCHED_OFFSET               0x800014
#define PORT_REG_SCHED_STRIDE               0x800018
#define PORT_REG_SCHED_TYPE                 0x80001C
#define PORT_REG_SCHED_ENABLE               0x800040
#define PORT_REG_RSS_MASK                   0x800080

94
95
#define PORT_QUEUE_ENABLE   0x900000

Jialin Li's avatar
Jialin Li committed
96
97
namespace corundum {

Jialin Li's avatar
Jialin Li committed
98
#define DESC_SIZE 16
Jialin Li's avatar
Jialin Li committed
99
100
#define CPL_SIZE 32
#define EVENT_SIZE 32
Jialin Li's avatar
Jialin Li committed
101
102
103
#define MAX_DMA_LEN 2048

class DescRing;
Jialin Li's avatar
Jialin Li committed
104
105
106
107
108
109
110
111

struct Desc {
    uint16_t rsvd0;
    uint16_t tx_csum_cmd;
    uint32_t len;
    uint64_t addr;
} __attribute__((packed)) ;

Jialin Li's avatar
Jialin Li committed
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
struct Cpl {
    uint16_t queue;
    uint16_t index;
    uint16_t len;
    uint16_t rsvd0;
    uint32_t ts_ns;
    uint16_t ts_s;
    uint16_t rx_csum;
    uint32_t rx_hash;
    uint8_t rx_hash_type;
    uint8_t rsvd1;
    uint8_t rsvd2;
    uint8_t rsvd3;
    uint32_t rsvd4;
    uint32_t rsvd5;
} __attribute__((packed)) ;

#define EVENT_TYPE_TX_CPL 0x0000
#define EVENT_TYPE_RX_CPL 0x0001

struct Event {
    uint16_t type;
    uint16_t source;
} __attribute__((packed)) ;
Jialin Li's avatar
Jialin Li committed
136

Jialin Li's avatar
Jialin Li committed
137
138
139
140
141
142
143
144
145
146
struct RxData {
    size_t len;
    uint8_t data[MAX_DMA_LEN];
};

#define DMA_TYPE_DESC   0
#define DMA_TYPE_MEM    1
#define DMA_TYPE_TX_CPL 2
#define DMA_TYPE_RX_CPL 3
#define DMA_TYPE_EVENT  4
Jialin Li's avatar
Jialin Li committed
147
148
149
150
151
152

struct DMAOp {
    uint8_t type;
    addr_t dma_addr;
    size_t len;
    DescRing *ring;
Jialin Li's avatar
Jialin Li committed
153
    RxData *rx_data;
Jialin Li's avatar
Jialin Li committed
154
155
156
157
158
    uint64_t tag;
    bool write;
    uint8_t data[MAX_DMA_LEN];
};

Jialin Li's avatar
Jialin Li committed
159
160
161
162
163
164
165
166
class DescRing {
public:
    DescRing();
    ~DescRing();

    addr_t dmaAddr();
    size_t sizeLog();
    unsigned index();
167
168
    ptr_t headPtr();
    ptr_t tailPtr();
Jialin Li's avatar
Jialin Li committed
169
170
171
172
173

    void setDMALower(uint32_t addr);
    void setDMAUpper(uint32_t addr);
    void setSizeLog(size_t size_log);
    void setIndex(unsigned index);
174
175
    virtual void setHeadPtr(ptr_t ptr);
    void setTailPtr(ptr_t ptr);
Jialin Li's avatar
Jialin Li committed
176

Jialin Li's avatar
Jialin Li committed
177
    virtual void dmaDone(DMAOp *op) = 0;
Jialin Li's avatar
Jialin Li committed
178

Jialin Li's avatar
Jialin Li committed
179
protected:
Jialin Li's avatar
Jialin Li committed
180
    bool empty();
Jialin Li's avatar
Jialin Li committed
181
    bool full();
Jialin Li's avatar
Jialin Li committed
182

Jialin Li's avatar
Jialin Li committed
183
184
185
    addr_t _dmaAddr;
    size_t _sizeLog;
    size_t _size;
Jialin Li's avatar
Jialin Li committed
186
    size_t _sizeMask;
Jialin Li's avatar
Jialin Li committed
187
    unsigned _index;
188
189
    ptr_t _headPtr;
    ptr_t _tailPtr;
Jialin Li's avatar
Jialin Li committed
190
    ptr_t _currHead;
191
    ptr_t _currTail;
Jialin Li's avatar
Jialin Li committed
192
193
194
    bool active;
};

Jialin Li's avatar
Jialin Li committed
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
class EventRing : public DescRing {
public:
    EventRing();
    ~EventRing();

    virtual void dmaDone(DMAOp *op) override;
    void issueEvent(unsigned type, unsigned source);
};

class CplRing : public DescRing {
public:
    CplRing(EventRing *eventRing);
    ~CplRing();

    virtual void dmaDone(DMAOp *op) override;
Jialin Li's avatar
Jialin Li committed
210
    void complete(unsigned index, size_t len, bool tx);
Jialin Li's avatar
Jialin Li committed
211
212
213
214
215

private:
    EventRing *eventRing;
};

Jialin Li's avatar
Jialin Li committed
216
217
class TxRing : public DescRing {
public:
Jialin Li's avatar
Jialin Li committed
218
    TxRing(CplRing *cplRing);
Jialin Li's avatar
Jialin Li committed
219
220
    ~TxRing();

221
    virtual void setHeadPtr(ptr_t ptr) override;
Jialin Li's avatar
Jialin Li committed
222
    virtual void dmaDone(DMAOp *op) override;
Jialin Li's avatar
Jialin Li committed
223
224
225

private:
    CplRing *txCplRing;
Jialin Li's avatar
Jialin Li committed
226
227
};

Jialin Li's avatar
Jialin Li committed
228
229
230
231
232
233
234
235
236
237
238
239
class RxRing : public DescRing {
public:
    RxRing(CplRing *cplRing);
    ~RxRing();

    virtual void dmaDone(DMAOp *op) override;
    void rx(RxData *rx_data);

private:
    CplRing *rxCplRing;
};

Jialin Li's avatar
Jialin Li committed
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
class Port {
public:
    Port();
    ~Port();

    unsigned id();
    unsigned features();
    size_t mtu();
    size_t schedCount();
    addr_t schedOffset();
    addr_t schedStride();
    unsigned schedType();
    unsigned rssMask();

    void setId(unsigned id);
    void setFeatures(unsigned features);
    void setMtu(size_t mtu);
    void setSchedCount(size_t count);
    void setSchedOffset(addr_t offset);
    void setSchedStride(addr_t stride);
    void setSchedType(unsigned type);
    void setRssMask(unsigned mask);
    void schedEnable();
    void schedDisable();
264
265
    void queueEnable();
    void queueDisable();
Jialin Li's avatar
Jialin Li committed
266
267
268
269
270
271
272
273
274
275
276

private:
    unsigned _id;
    unsigned _features;
    size_t _mtu;
    size_t _schedCount;
    addr_t _schedOffset;
    addr_t _schedStride;
    unsigned _schedType;
    unsigned _rssMask;
    bool _schedEnable;
277
    bool _queueEnable;
Jialin Li's avatar
Jialin Li committed
278
279
};

Jialin Li's avatar
Jialin Li committed
280
281
282
283
284
285
286
class Corundum {
public:
    Corundum();
    ~Corundum();

    reg_t readReg(addr_t addr);
    void writeReg(addr_t addr, reg_t val);
Jialin Li's avatar
Jialin Li committed
287
    void rx(uint8_t port, RxData *rx_data);
Jialin Li's avatar
Jialin Li committed
288
289

private:
Jialin Li's avatar
Jialin Li committed
290
    EventRing eventRing;
Jialin Li's avatar
Jialin Li committed
291
    TxRing txRing;
Jialin Li's avatar
Jialin Li committed
292
    CplRing txCplRing;
Jialin Li's avatar
Jialin Li committed
293
    RxRing rxRing;
Jialin Li's avatar
Jialin Li committed
294
    CplRing rxCplRing;
Jialin Li's avatar
Jialin Li committed
295
    Port port;
Jialin Li's avatar
Jialin Li committed
296
};
Jialin Li's avatar
Jialin Li committed
297
298

} // namespace corundum