异构计算关键技术之多线程技术（四）
最近遇到了一个项目，需要写一个用户态的测试程序（独立进程），用来测试FPGA PCIe DMA的性能，具体的要求如下：
1. 需要一个主线程，用来配置FPGA的寄存器，同时启动从线程；
2. 如果不进行人为干涉，子线程一直进行FPGA的相关操作，比如下发trigger信号、配置burst次数、数据长度；
3. 进行人为干涉，子线程退出，并返回子线程执行的一些信息，提供给主线程做统计和计算信息；
下面我们直接给出相关的核心代码，结合线程的理论进行分析：
...
...
static int run = 0;
static int round = 0;
...
typedef struct _param
{
struct util_mem *util;
int burst;
int len;
}param;
void *
recv_perf(void *data)
{
...
cpu_set_t mask;
CPU_ZERO(&mask);
CPU_SET(14, &mask);
sched_setaffinity(0, sizeof(cpu_set_t), &mask);
...
/* len and burst*/
reg_write(..., addr, (p->len&0x0000ffff)|((p->burst&0x0000ffff)<<16)));
...
while(run) {
/* trigger */
reg_write(..., addr, &rdata);
while(times < p->burst) {
data_size = recv(...,...,...);
if (data_size == xxx) {
...
times++;
}
}
times = 0;
cnt++;
}
round = cnt * p->burst;
pthread_exit(0);
}
int main(int argc, char **argv)
{
...
pthread_t tid;
ret = start(...);
ret = pthread_create(&tid, NULL, recv_perf, &data);
if (ret < 0) {
}
else {
}
pthread_detach(tid);
while(1) {
ch = getchar();
if (ch = 's') {
run = 0;
...
break;
}
}
...
avg = (float)recv_total/round;
...
}
<font color=B871F78><h2>一、代码设计分析 </font></h2>
这段代码非常实用，整体思想如下：
-
主线程main函数，首先做了FPGA系统的一些初始化功能，然后起了一个从线程recv_perf();
-
从线程主要是根据传递的参数发送给FPGA，让FPGA一直做DMA操作；
-
主从线程分离detach();
-
主线程while(1)循环，用来控制从线程的结束，同时通过全局变量进行传递参数；
-
最后计算FPGA的统计信息；
1710239945977.png
二、C++多线程编程知识点归纳
1. 主线程和子线程的区别
我们先看看线程是如何创建起来的：
进程仅仅是一个容器，包含了线程运行中所需要的数据结构等信息。
一个进程创建时，操作系统会创建一个线程，这就是主线程。
而其他的从线程，却要主线程的代码来创建，也就是由程序员来创建。
主线程：
main（）函数均视为主线程，除了“不包含在thread里面的程序”，均视为主线程；
子线程：
包含在thread = new thread（）里面均视为子线程；
main函数：
main()函数作为入口开始运行，是一个进程，同时也是一个线程。在现在的操作系统中，都是多线程的。
2. 线程的创建与参数传递
这个实例中，我们需要做一个子线程，用来一直执行FPGA的操作，同时我们需要传递FPGA的配置参数，下发给FPGA寄存器空间。
linux下的多线程程序，需要使用pthread.h，链接时需要使用libthread.a。
线程的创建需要通过pthread_create来完成，声明如下：
#include <pthread.h>
int pthread_create(pthread_t *thread, pthread_attr_t *attr, void* (start_routine)(void*), void *arg);
- thread：是一个指针，线程创建成功时，用以返回创建的线程ID；
- attr：线程属性，NULL表示使用默认；
- start_rountine：函数指针，指被创建的线程函数；
- arg：该参数指向传递给线程函数的参数；
实例中，接收函数recv_perf()，同时传递的参数结构体data；
3. 线程的退出
多线程中，终止执行的方式有3种，分别是：
1. 线程执行完成后，自行终止；
2. 线程执行种，遇到了pthread_exit()或者return；
3. 线程在执行过程种，接收到了其他线程发送的“终止执行”的信息，然后终止执行；
第一种很容易理解，不做讨论。
pthread_exit()和return：
return：
return 关键字用于终止函数执行，必要时还能将函数的执行结果反馈给调用者。
return 关键字不仅可以用于普通函数，线程函数中也可以使用它。
pthread_exit()：
<pthread.h>头文件中，提供有一个和 return 关键字相同功能的 pthread_exit() 函数。
和之前不同，pthread_exit() 函数只适用于线程函数，而不能用于普通函数。
void pthread_exit(void*retval);
retval是void*类型的指针，可以指向任何类型的数据，它指向的数据作为线程退出的返回值。
pthread_exit()和return()的区别：
- return：不仅会终止主线程执行，还会终止其他子线程的执行；
- pthread_exit()：只会终止当前线程，不会影响到其他线程的执行；
实际场景中，想要终止某个子线程，强烈建议使用pthread_exit()函数。
pthread_cancel:
一个线程还可以向另一个线程发送“终止执行”的信号（后续称为“cancel”信号），这时候需要调用pthread_cancel()函数。
int pthread_cancel(pthread_t thread);
参数thread用于接收cancel信号的目标线程。
对于接收cancel信号后，结束执行的目标线程，等同于该线程自己执行如下语句：
pthread_exit(PTHREAD_CANCELED);
也就是说，当一个线程被强制终止时，它会返回pthread_cancel这个宏。
然后对于我们这个设计，巧妙的使用了run这个全局变量，用来控制子线程执行，同时利用全局变量来进行计算，是个很好的策略。
这是因为子线程在detach()以后，就无法再返回子线程的资源，会出现core。
4. detach()
detach()的作用是将子线程和主线程的关联分离，也就是说detach()后子线程在后台独立继续执行，主线程无法再获得子线程的控制权。
即使主线程结束，子线程未执行也不会结束。当主线程结束时，由运行时库负责清理和子线程相关的资源。
detach()同时也带来了一些问题，如子线程要访问主线程的对象，而主线中的对象又因为主线程结束而被销毁，导致程序崩溃。
5. 把进程/线程绑定到特定的cpu核上运行
某个进程需要较高的运行效率时，就有必要考虑将其绑定到单独的核上运行，以减小由于在不同的核上调度造成的开销。
把某个进程/线程绑定到特定的cpu核上后，该进程就会一直在此核上运行，不会再被操作系统调度到其他核上。但绑定的这个核还是可能会被调度运行其他应用程序的。（可以做隔离）
查看绑定情况：
taskset -p pid
显示的是十进制，需要转换成2进制，每个1对应一个cpu（cpu从0开始）
启动时绑定：
taskset -c xxx,yyy ./pcie_perf&
启动应用程序的时候绑定。
启动后绑定：
taskset -cp 1,2,5,11 9865 将进程9864绑定到#1、#2、#5、#11号核上面。
taskset -cp 1,2,5-11 9865 将进程9864绑定到#1、#2、#5~#11号核上面。
代码绑定：
...
cpu_set_t mask;
CPU_ZERO(&mask);
CPU_SET(14, &mask);
sched_setaffinity(0, sizeof(cpu_set_t), &mask);
...
三、未完待续
欢迎关注知乎：北京不北，+vbeijing_bubei
欢迎+V：beijing_bubei
欢迎关注douyin：near.X （北京不北）
获得免费答疑，长期技术交流。
网友评论