ivampiresp/Spdk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lib/thread: Add spdk_thread_set_cpumask() and enum SPDK_THREAD_OP_RESCHED

Browse Source 
Add an new API spdk_thread_set_cpumask() and enum SPDK_THREAD_OP_RESCHED.

spdk_thread_set_cpumask() can be called only from the current thread
and requires SPDK thread operation supports reschedule operation.

spdk_thread_set_cpumask() updates the cpumask of the current thread to
the specified value, and then invokes framework's reschedule operation
to the thread.

If spdk_thread_set_cpumask() calls multiple times in a single
spdk_thread_poll() context, the last value will be used in the
reschedule operation.

Signed-off-by: Shuhei Matsumoto <shuhei.matsumoto.xt@hitachi.com>
Change-Id: I7808626b10269543c1e2cd86793a504daa4b6389
Reviewed-on: https://review.spdk.io/gerrit/c/spdk/spdk/+/499
Tested-by: SPDK CI Jenkins <sys_sgci@intel.com>
Reviewed-by: Ben Walker <benjamin.walker@intel.com>
Reviewed-by: Jim Harris <james.r.harris@intel.com>
This commit is contained in:
Shuhei Matsumoto 2020-02-23 18:02:53 -05:00 committed by Tomasz Zawadzki
parent 1368aec870
commit 713aafbc10
2 changed files with 46 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
include/spdk/thread.h
View File

@ -76,6 +76,11 @@ enum spdk_thread_op {
* should frequently call spdk_thread_poll() on the thread provided. * should frequently call spdk_thread_poll() on the thread provided.
*/ */
SPDK_THREAD_OP_NEW, SPDK_THREAD_OP_NEW,
/* Called when SPDK thread needs to be rescheduled. (e.g., when cpumask of the
* SPDK thread is updated.
*/
SPDK_THREAD_OP_RESCHED,
}; };
/** /**
@ -309,6 +314,18 @@ void *spdk_thread_get_ctx(struct spdk_thread *thread);
*/ */
struct spdk_cpuset *spdk_thread_get_cpumask(struct spdk_thread *thread); struct spdk_cpuset *spdk_thread_get_cpumask(struct spdk_thread *thread);
/**
* Set the current thread's cpumask to the specified value. The thread may be
* rescheduled to one of the CPUs specified in the cpumask.
*
* This API requires SPDK thread operation supports SPDK_THREAD_OP_RESCHED.
*
* \param cpumask The new cpumask for the thread.
*
* \return 0 on success, negated errno otherwise.
*/
int spdk_thread_set_cpumask(struct spdk_cpuset *cpumask);
/** /**
* Return the thread object associated with the context handle previously * Return the thread object associated with the context handle previously
* obtained by calling spdk_thread_get_ctx(). * obtained by calling spdk_thread_get_ctx().

29
lib/thread/thread.c
View File

@ -479,6 +479,35 @@ spdk_thread_get_cpumask(struct spdk_thread *thread)
return &thread->cpumask; return &thread->cpumask;
} }
int
spdk_thread_set_cpumask(struct spdk_cpuset *cpumask)
{
struct spdk_thread *thread;
if (!g_thread_op_supported_fn || !g_thread_op_supported_fn(SPDK_THREAD_OP_RESCHED)) {
SPDK_ERRLOG("Framework does not support reschedule operation.\n");
assert(false);
return -ENOTSUP;
}
thread = spdk_get_thread();
if (!thread) {
SPDK_ERRLOG("Called from non-SPDK thread\n");
assert(false);
return -EINVAL;
}
spdk_cpuset_copy(&thread->cpumask, cpumask);
/* Invoke framework's reschedule operation. If this function is called multiple times
* in a single spdk_thread_poll() context, the last cpumask will be used in the
* reschedule operation.
*/
g_thread_op_fn(thread, SPDK_THREAD_OP_RESCHED);
return 0;
}
struct spdk_thread * struct spdk_thread *
spdk_thread_get_from_ctx(void *ctx) spdk_thread_get_from_ctx(void *ctx)
{ {