OrgLion-Writes/DemoPublic
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

修正 SetThreadAffinity 接口,调整代码格式.

Browse Source 
Signed-off-by: rick.chan <chenyang@autoai.com>
This commit is contained in:
rick.chan 2021-06-07 16:50:29 +08:00
parent bba3271108
commit f498e91191
1 changed files with 44 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
Android/Native/CpuAffinity/main.cpp
View File

@ -32,21 +32,23 @@ int SetProcessAffinity(void) {
return res; return res;
} }
void SetThreadAffinity(int mask) { void SetThreadAffinity(size_t cpusetsize, cpu_set_t *cpuset) {
int err, syscallres; int err, syscallres;
pid_t pid = gettid(); pid_t pid = gettid();
// Linux 下可以直接使用 pthread_setaffinity_np() 函数将当前线程绑定在具体的 CPU 上, 而 Android 屏蔽了该 API, 需要通过 syscall(__NR_sched_setaffinity) 来实现. // Linux 下可以直接使用 pthread_setaffinity_np() 函数将当前线程绑定在具体的 CPU 上, 而 Android 屏蔽了该 API, 需要通过 syscall(__NR_sched_setaffinity) 来实现.
syscallres = syscall(__NR_sched_setaffinity, pid, sizeof(mask), &mask); syscallres = syscall(__NR_sched_setaffinity, pid, cpusetsize, cpuset);
if (syscallres) { if(syscallres)
err = errno; err = errno;
}
} }
void* DemoRoutine0(void* arg) void* DemoRoutine0(void* arg)
{ {
(void)arg; (void)arg;
cpu_set_t set;
CPU_ZERO(&set);
// DemoRoutine0 只运行在核 3 上. // DemoRoutine0 只运行在核 3 上.
SetThreadAffinity(1<<3); CPU_SET(3, &set);
SetThreadAffinity(sizeof(set), &set);
while(1); while(1);
return nullptr; return nullptr;
} }
@ -54,8 +56,11 @@ void* DemoRoutine0(void* arg)
void* DemoRoutine1(void* arg) void* DemoRoutine1(void* arg)
{ {
(void)arg; (void)arg;
cpu_set_t set;
CPU_ZERO(&set);
// DemoRoutine0 只运行在核 7 上. // DemoRoutine0 只运行在核 7 上.
SetThreadAffinity(1<<7); CPU_SET(7, &set);
SetThreadAffinity(sizeof(set), &set);
while(1); while(1);
return nullptr; return nullptr;
} }
@ -65,11 +70,9 @@ int main(void)
pthread_t demoThrd0, demoThrd1; pthread_t demoThrd0, demoThrd1;
cpu_set_t set; cpu_set_t set;
sched_getaffinity(getpid(), sizeof(set), &set);
printf("Init CPU Setbits=%lu.\r\n", set.__bits[0]);
SetProcessAffinity(); SetProcessAffinity();
sched_getaffinity(getpid(), sizeof(set), &set); sched_getaffinity(getpid(), sizeof(set), &set);
printf("Init CPU Setbits=%lu.\r\n", set.__bits[0]); printf("Current CPU Setbits=%lu.\r\n", set.__bits[0]);
pthread_create(&demoThrd0, nullptr, DemoRoutine0, nullptr); pthread_create(&demoThrd0, nullptr, DemoRoutine0, nullptr);
pthread_create(&demoThrd1, nullptr, DemoRoutine1, nullptr); pthread_create(&demoThrd1, nullptr, DemoRoutine1, nullptr);