xref: /dpdk/lib/sched/rte_pie.c (revision b53d106d34b5c638f5a2cbdfee0da5bd42d4383f)
1 /* SPDX-License-Identifier: BSD-3-Clause
2  * Copyright(c) 2020 Intel Corporation
3  */
4 
5 #include <stdlib.h>
6 
7 #include "rte_pie.h"
8 #include <rte_common.h>
9 #include <rte_cycles.h>
10 #include <rte_malloc.h>
11 
12 #ifdef __INTEL_COMPILER
13 #pragma warning(disable:2259) /* conversion may lose significant bits */
14 #endif
15 
16 int
17 rte_pie_rt_data_init(struct rte_pie *pie)
18 {
19 	if (pie == NULL) {
20 		/* Allocate memory to use the PIE data structure */
21 		pie = rte_malloc(NULL, sizeof(struct rte_pie), 0);
22 
23 		if (pie == NULL)
24 			RTE_LOG(ERR, SCHED, "%s: Memory allocation fails\n", __func__);
25 
26 		return -1;
27 	}
28 
29 	pie->active = 0;
30 	pie->in_measurement = 0;
31 	pie->departed_bytes_count = 0;
32 	pie->start_measurement = 0;
33 	pie->last_measurement = 0;
34 	pie->qlen = 0;
35 	pie->avg_dq_time = 0;
36 	pie->burst_allowance = 0;
37 	pie->qdelay_old = 0;
38 	pie->drop_prob = 0;
39 	pie->accu_prob = 0;
40 
41 	return 0;
42 }
43 
44 int
45 rte_pie_config_init(struct rte_pie_config *pie_cfg,
46 	const uint16_t qdelay_ref,
47 	const uint16_t dp_update_interval,
48 	const uint16_t max_burst,
49 	const uint16_t tailq_th)
50 {
51 	uint64_t tsc_hz = rte_get_tsc_hz();
52 
53 	if (pie_cfg == NULL)
54 		return -1;
55 
56 	if (qdelay_ref <= 0) {
57 		RTE_LOG(ERR, SCHED,
58 			"%s: Incorrect value for qdelay_ref\n", __func__);
59 		return -EINVAL;
60 	}
61 
62 	if (dp_update_interval <= 0) {
63 		RTE_LOG(ERR, SCHED,
64 			"%s: Incorrect value for dp_update_interval\n", __func__);
65 		return -EINVAL;
66 	}
67 
68 	if (max_burst <= 0) {
69 		RTE_LOG(ERR, SCHED,
70 			"%s: Incorrect value for max_burst\n", __func__);
71 		return -EINVAL;
72 	}
73 
74 	if (tailq_th <= 0) {
75 		RTE_LOG(ERR, SCHED,
76 			"%s: Incorrect value for tailq_th\n", __func__);
77 		return -EINVAL;
78 	}
79 
80 	pie_cfg->qdelay_ref = (tsc_hz * qdelay_ref) / 1000;
81 	pie_cfg->dp_update_interval = (tsc_hz * dp_update_interval) / 1000;
82 	pie_cfg->max_burst = (tsc_hz * max_burst) / 1000;
83 	pie_cfg->tailq_th = tailq_th;
84 
85 	return 0;
86 }
87