Add new get/set APIs to configure graph worker model which is used to
determine which model will be chosen.
Signed-off-by: Haiyue Wang <haiyue.w...@intel.com>
Signed-off-by: Cunming Liang <cunming.li...@intel.com>
Signed-off-by: Zhirun Yan <zhirun....@intel.com>
---
lib/graph/meson.build | 1 +
lib/graph/rte_graph_worker.c | 21 ++++++++++
lib/graph/rte_graph_worker_common.h | 61 +++++++++++++++++++++++++++++
lib/graph/version.map | 3 ++
4 files changed, 86 insertions(+)
create mode 100644 lib/graph/rte_graph_worker.c
diff --git a/lib/graph/meson.build b/lib/graph/meson.build
index 3526d1b5d4..9fab8243da 100644
--- a/lib/graph/meson.build
+++ b/lib/graph/meson.build
@@ -15,6 +15,7 @@ sources = files(
'graph_stats.c',
'graph_populate.c',
'graph_pcap.c',
+ 'rte_graph_worker.c',
)
headers = files('rte_graph.h', 'rte_graph_worker.h')
diff --git a/lib/graph/rte_graph_worker.c b/lib/graph/rte_graph_worker.c
new file mode 100644
index 0000000000..3a4215f1a2
--- /dev/null
+++ b/lib/graph/rte_graph_worker.c
@@ -0,0 +1,21 @@
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(C) 2023 Intel Corporation
+ */
+
+#include "rte_graph_worker_common.h"
+#include "graph_private.h"
+
+int
+rte_graph_worker_model_set(uint32_t model)
+{
+ struct graph_head *graph_head = graph_list_head_get();
+ struct graph *graph;
+
+ if (graph_model_is_valid(model))
+ return -EINVAL;
+
+ STAILQ_FOREACH(graph, graph_head, next)
+ graph->graph->model = model;
+
+ return 0;
+}
diff --git a/lib/graph/rte_graph_worker_common.h
b/lib/graph/rte_graph_worker_common.h
index 41428974db..5dba3c0edd 100644
--- a/lib/graph/rte_graph_worker_common.h
+++ b/lib/graph/rte_graph_worker_common.h
@@ -29,6 +29,13 @@
extern "C" {
#endif
+/** Graph worker models */
+/* If adding new entry, then update graph_model_is_valid API. */
+#define RTE_GRAPH_MODEL_RTC 0 /**< Run-To-Completion model. It is the default
model. */
+#define RTE_GRAPH_MODEL_DEFAULT RTE_GRAPH_MODEL_RTC /**< Default graph model.
*/
+#define RTE_GRAPH_MODEL_MCORE_DISPATCH 1
+/**< Dispatch model to support cross-core dispatching within core affinity. */
+
/**
* @internal
*
@@ -41,6 +48,7 @@ struct rte_graph {
rte_node_t nb_nodes; /**< Number of nodes in the graph. */
rte_graph_off_t *cir_start; /**< Pointer to circular buffer. */
rte_graph_off_t nodes_start; /**< Offset at which node memory starts. */
+ uint32_t model; /**< graph model */
rte_graph_t id; /**< Graph identifier. */
int socket; /**< Socket ID where memory is allocated. */
char name[RTE_GRAPH_NAMESIZE]; /**< Name of the graph. */
@@ -490,6 +498,59 @@ rte_node_next_stream_move(struct rte_graph *graph, struct
rte_node *src,
}
}
+/**
+ * Test the validity of model.
+ *
+ * @param id
+ * Node id to check.
+ *
+ * @return
+ * true if graph model is valid, false otherwise.
+ */
+static __rte_always_inline
+bool
+graph_model_is_valid(uint32_t model)
+{
+ if (model > RTE_GRAPH_MODEL_MCORE_DISPATCH)
+ return false;
+
+ return true;
+}
+
+/**
+ * @note This function does not perform any locking, and is only safe to call
+ * before graph running. It will set all graphs the same model.
+ *
+ * @param model
+ * Name of the graph worker model.
+ *
+ * @return
+ * 0 on success, -1 otherwise.
+ */
+__rte_experimental
+int rte_graph_worker_model_set(uint32_t model);
+
+/**
+ * Get the graph worker model
+ *
+ * @note All graph will use the same model and this function will get model
from the first one
+ *
+ * @param graph
+ * Graph pointer.
+ *
+ * @return
+ * Graph worker model on success.
+ */
+__rte_experimental
+static inline uint32_t
+rte_graph_worker_model_get(struct rte_graph *graph)
+{
+ if (!graph_model_is_valid(graph->model))
+ return -EINVAL;
+
+ return graph->model;
+}
+
#ifdef __cplusplus
}
#endif
diff --git a/lib/graph/version.map b/lib/graph/version.map
index 13b838752d..eea73ec9ca 100644
--- a/lib/graph/version.map
+++ b/lib/graph/version.map
@@ -43,5 +43,8 @@ EXPERIMENTAL {
rte_node_next_stream_put;
rte_node_next_stream_move;
+ rte_graph_worker_model_set;
+ rte_graph_worker_model_get;
+
local: *;
};
--
2.37.2
