乐趣区

关于openmp:OpenMP-Parallel-Construct-实现原理与源码分析

OpenMP Parallel Construct 实现原理与源码剖析

前言

在本篇文章当中咱们将次要剖析 OpenMP 当中的 parallel construct 具体时如何实现的,以及这个 construct 调用了哪些运行时库函数,并且详细分析这期间的参数传递!

Parallel 剖析——编译器角度

在本大节当中咱们将从编译器的角度去剖析该如何解决 parallel construct。首先从词法剖析和语法分析的角度来说这对编译器并不难,只须要加上一些解决规定,要害是编译器将一个 parallel construct 具体编译成了什么?

上面是一个非常简单的 parallel construct。

#pragma omp parallel
{body;}

编译器在遇到下面的 parallel construct 之后会将代码编译成上面的样子:

void subfunction (void *data)
{
  use data;
  body;
}

setup data;
GOMP_parallel_start (subfunction, &data, num_threads);
subfunction (&data);
GOMP_parallel_end ();

首先 parallel construct 中的代码块会被编译成一个函数 sub function,当然了函数名不肯定是这个,而后会在应用 #pragma omp parallel 的函数当中将一个 parallel construct 编译成 OpenMP 动静库函数的调用,在下面的伪代码当中也指出了,具体会调用 OpenMP 的两个库函数 GOMP_parallel_start 和 GOMP_parallel_end,并且主线程也会调用函数 subfunction,咱们在前面的文章当中在仔细分析这两个动静库函数的源代码。

深刻分析 Parallel 动静库函数参数传递

动静库函数剖析

在本大节当中,咱们次要去剖析一下在 OpenMP 当中共享参数是如何传递的,以及介绍函数 GOMP_parallel_start 的几个参数的含意。

首先咱们剖析函数 GOMP_parallel_start 的参数含意,这个函数的函数原型如下:

void GOMP_parallel_start (void (*fn)(void *), void *data, unsigned num_threads)

下面这个函数一共有三个参数:

  • 第一个参数 fn 是一个函数指针,次要是用于指向下面编译进去的 subfunction 这个函数的,因为须要多个线程同时执行这个函数,因而须要将这个函数传递过来,让不同的线程执行。
  • 第二个参数是传递的数据,咱们在并行域当中会应用到共享的或者公有的数据,这个指针次要是用于传递数据的,咱们在前面会仔细分析这个参数的应用。
  • 第三个参数是示意 num_threads 子句指定的线程个数,如果不指定这个子句默认的参数是 0,然而如果你应用了 IF 子句并且条件是 false 的话,那么这个参数的值就是 1。
  • 这个函数的次要作用是启动一个或者多个线程,并且执行函数 fn。
void GOMP_parallel_end (void)
  • 这个函数的次要作用是进行线程的同步,因为一个 parallel 并行域须要期待所有的线程都执行实现之后才持续往后执行。除此之外还须要开释线程组的资源并行返回到之前的 omp_in_parallel() 示意的状态。

参数传递剖析

咱们当初应用上面的代码来具体分析参数传递过程:

#include <stdio.h>
#include "omp.h"

int main()
{
  int data = 100;
  int two  = -100;
  printf("start\n");
#pragma omp parallel num_threads(4) default(none) shared(data, two)
  {printf("tid = %d data = %d two = %d\n", omp_get_thread_num(), data, two);
  }

  printf("finished\n");
  return 0;
}

咱们首先来剖析一下下面的两个变量 data 和 two 的是如何被传递的,咱们首先用图的形式进行示意,而后剖析一下汇编程序并且对图进行验证。

下面的代码当中两个变量 datatwo 在内存当中的布局构造大抵如下所示(假如 data 的初始地位时 0x0):

那么在函数 GOMP_parallel_start 当中传递的参数 data 就是 0x0 也就是指向 data 的内存地址,如下图所示:

那么依据下面参数传递的状况,咱们就能够在 subfunction 当中应用 *(int*)data 失去 data 的值,应用 *((int*) ((char*)data + 4)) 失去 two 的值,如果是 private 传递的话咱们就能够先拷贝这个数据再应用,如果是 shared 的话,那么咱们就能够间接应用指针就行啦。

下面的程序咱们用 pthread 大抵形容一下,则 pthread 对应的代码如下所示:


#include "pthread.h"
#include "stdio.h"
#include "stdint.h"

typedef struct data_in_main_function{
    int data;
    int two;
}data_in_main_function;

pthread_t threads[4];

void* subfunction(void* data)
{int two = ((data_in_main_function*)data)->two;
  int data_ = ((data_in_main_function*)data)->data;
  printf("tid = %ld data = %d two = %d\n", pthread_self(), data_, two);
  return NULL;
}

int main()
{
  // 在主函数申请 8 个字节的栈空间
  data_in_main_function data;
  data.data = 100;
  data.two = -100;
  for(int i = 0; i < 4; ++i)
  {pthread_create(&threads[i], NULL, subfunction, &data);
  }
  for(int i = 0; i < 4; ++i)
  {pthread_join(threads[i], NULL);
  }
  return 0;
}

汇编程序剖析

在本节当中咱们将认真去剖析下面的程序所产生的汇编程序,在本文当中的汇编程序根底 x86_64 平台。在剖析汇编程序之前咱们首先须要理解一下 x86 函数的调用规约,具体来说就是在进行函数调用的时候哪些寄存器保留函数参数以及是第几个函数参数。具体的规定如下所示:

寄存器 含意
rdi 第一个参数
rsi 第二个参数
rdx 第三个参数
rcx 第四个参数
r8 第五个参数
r9 第六个参数

咱们当初仔细分析一下下面的程序的 main 函数的反汇编程序:

00000000004006cd <main>:
  4006cd:       55                      push   %rbp
  4006ce:       48 89 e5                mov    %rsp,%rbp
  4006d1:       48 83 ec 10             sub    $0x10,%rsp
  4006d5:       c7 45 fc 64 00 00 00    movl   $0x64,-0x4(%rbp)
  4006dc:       c7 45 f8 9c ff ff ff    movl   $0xffffff9c,-0x8(%rbp)
  4006e3:       bf f4 07 40 00          mov    $0x4007f4,%edi
  4006e8:       e8 93 fe ff ff          callq  400580 <puts@plt>
  4006ed:       8b 45 fc                mov    -0x4(%rbp),%eax
  4006f0:       89 45 f0                mov    %eax,-0x10(%rbp)
  4006f3:       8b 45 f8                mov    -0x8(%rbp),%eax
  4006f6:       89 45 f4                mov    %eax,-0xc(%rbp)
  4006f9:       48 8d 45 f0             lea    -0x10(%rbp),%rax
  4006fd:       ba 04 00 00 00          mov    $0x4,%edx
  400702:       48 89 c6                mov    %rax,%rsi
  400705:       bf 3d 07 40 00          mov    $0x40073d,%edi
  40070a:       e8 61 fe ff ff          callq  400570 <GOMP_parallel_start@plt>
  40070f:       48 8d 45 f0             lea    -0x10(%rbp),%rax
  400713:       48 89 c7                mov    %rax,%rdi
  400716:       e8 22 00 00 00          callq  40073d <main._omp_fn.0>
  40071b:       e8 70 fe ff ff          callq  400590 <GOMP_parallel_end@plt>
  400720:       8b 45 f0                mov    -0x10(%rbp),%eax
  400723:       89 45 fc                mov    %eax,-0x4(%rbp)
  400726:       8b 45 f4                mov    -0xc(%rbp),%eax
  400729:       89 45 f8                mov    %eax,-0x8(%rbp)
  40072c:       bf fa 07 40 00          mov    $0x4007fa,%edi
  400731:       e8 4a fe ff ff          callq  400580 <puts@plt>
  400736:       b8 00 00 00 00          mov    $0x0,%eax
  40073b:       c9                      leaveq 
  40073c:       c3                      retq   

从下面的反汇编程序咱们能够看到在主函数的汇编代码当中的确调用了函数 GOMP_parallel_start 和 GOMP_parallel_end,并且 subfunction 为 main._omp_fn.0,它对应的汇编程序如下所示:

000000000040073d <main._omp_fn.0>:
  40073d:       55                      push   %rbp
  40073e:       48 89 e5                mov    %rsp,%rbp
  400741:       48 83 ec 10             sub    $0x10,%rsp
  400745:       48 89 7d f8             mov    %rdi,-0x8(%rbp)
  400749:       e8 52 fe ff ff          callq  4005a0 <omp_get_thread_num@plt>
  40074e:       48 8b 55 f8             mov    -0x8(%rbp),%rdx
  400752:       8b 4a 04                mov    0x4(%rdx),%ecx
  400755:       48 8b 55 f8             mov    -0x8(%rbp),%rdx
  400759:       8b 12                   mov    (%rdx),%edx
  40075b:       89 c6                   mov    %eax,%esi
  40075d:       bf 03 08 40 00          mov    $0x400803,%edi
  400762:       b8 00 00 00 00          mov    $0x0,%eax
  400767:       e8 44 fe ff ff          callq  4005b0 <printf@plt>
  40076c:       c9                      leaveq 
  40076d:       c3                      retq   
  40076e:       66 90                   xchg   %ax,%ax

GOMP_parallel_start 具体参数剖析

  • void (*fn)(void *),咱们当初来看一下函数 GOMP_parallel_start 的第一个参数,依据咱们后面谈到的第一个参数应该保留在 rdi 寄存器,咱们当初剖析一下在 main 函数的反汇编程序当中在调用函数 GOMP_parallel_start 之前 rdi 寄存器的值。咱们能够看到在 main 函数地位为 4006f8 的中央的指令 mov $0x40073d,%edi 能够看到 rdi 寄存器的值为 0x40073d(edi 寄存器是 rdi 寄存器的低 32 位),咱们能够看到 函数 main._omp_fn.0 的起始地址就是 0x40073d,因而咱们就能够在函数 GOMP_parallel_start 应用这个函数指针了,最终在启动的线程当中调用这个函数。
  • void *data,这是函数 GOMP_parallel_start 的第二个参数,依据后面的剖析第二个参数保留在 rsi 寄存器当中,我当初将 main 数当中和 rsi 相干的指令抉择进去:
00000000004006cd <main>:
  4006cd:       55                      push   %rbp
  4006ce:       48 89 e5                mov    %rsp,%rbp
  4006d1:       48 83 ec 10             sub    $0x10,%rsp
  4006d5:       c7 45 fc 64 00 00 00    movl   $0x64,-0x4(%rbp)
  4006dc:       c7 45 f8 9c ff ff ff    movl   $0xffffff9c,-0x8(%rbp)
  4006ed:       8b 45 fc                mov    -0x4(%rbp),%eax
  4006f0:       89 45 f0                mov    %eax,-0x10(%rbp)
  4006f3:       8b 45 f8                mov    -0x8(%rbp),%eax
  4006f6:       89 45 f4                mov    %eax,-0xc(%rbp)
  4006f9:       48 8d 45 f0             lea    -0x10(%rbp),%rax
  400702:       48 89 c6                mov    %rax,%rsi

下面的汇编程序的栈空间以及在调用函数之前 GOMP_parallel_start 局部寄存器的指向如下所示:

最终在调用函数 GOMP_parallel_start 之前 rsi 寄存器的指向如上图所示,上图当中 rsi 的指向的内存地址作为参数传递过来。依据上文谈到的 subfunction 中的参数能够晓得,在函数 main._omp_fn.0 当中的 rdi 寄存器(也就是第一个参数 *data)的值就是上图当中 rsi 寄存器指向的内存地址的值(事实上也就是 rsi 寄存器的值)。大家能够自行对照着函数 main._omp_fn.0 的汇编程序对 rdi 寄存器的应用就能够晓得这其中的参数传递的过程了。

  • unsigned num_threads,依据前文提到的保留第三个参数的寄存器是 rdx,在 main 函数的地位 4006fd 处,指令为 mov $0x4,%edx,这和咱们本人写的程序是统一的都是 4 (0x4)。

动静库函数源码剖析

GOMP_parallel_start 源码剖析

咱们首先来看一下函数 GOMP_parallel_start 的源代码:

void
GOMP_parallel_start (void (*fn) (void *), void *data, unsigned num_threads)
{num_threads = gomp_resolve_num_threads (num_threads, 0);
  gomp_team_start (fn, data, num_threads, gomp_new_team (num_threads));
}

在这里咱们对函数 gomp_team_start 进行剖析,其余两个函数 gomp_resolve_num_threads 和 gomp_new_team 只简略进行作用阐明,太粗疏的源码剖析其实是没有必要的,感兴趣的同学自行剖析即可,咱们只须要理解整个执行流程即可。

  • gomp_resolve_num_threads,这个函数的次要作用是最终确定须要几个线程去执行工作,因为咱们可能并没有应用 num_threads 子句,而且这个值和环境变量也有关系,因而须要对线程的个数进行确定。
  • gomp_new_team,这个函数的次要作用是创立蕴含 num_threads 个线程数据的线程组,并且对数据进行初始化操作。
  • gomp_team_start,这个函数的次要作用是启动 num_threads 个线程去执行函数 fn,这其中波及一些细节,比如说线程的亲和性(affinity)设置。

因为 gomp_team_start 的源代码太长了,这里只是节选局部源程序进行剖析:

  /* Launch new threads.  */
  for (; i < nthreads; ++i, ++start_data)
    {
      pthread_t pt;
      int err;

      start_data->fn = fn; // 这行代码就是将 subfunction 函数指针进行保留最终在函数  gomp_thread_start 当中进行调用
      start_data->fn_data = data; // 这里保留函数 subfunction 的函数参数
      start_data->ts.team = team; // 线程的所属组
      start_data->ts.work_share = &team->work_shares[0];
      start_data->ts.last_work_share = NULL;
      start_data->ts.team_id = i; // 线程的 id 咱们能够应用函数 omp_get_thread_num 失去这个值
      start_data->ts.level = team->prev_ts.level + 1;
      start_data->ts.active_level = thr->ts.active_level;
#ifdef HAVE_SYNC_BUILTINS
      start_data->ts.single_count = 0;
#endif
      start_data->ts.static_trip = 0;
      start_data->task = &team->implicit_task[i];
      gomp_init_task (start_data->task, task, icv);
      team->implicit_task[i].icv.nthreads_var = nthreads_var;
      start_data->thread_pool = pool;
      start_data->nested = nested;
            // 如果应用了线程的亲和性那么还须要进行亲和性设置
      if (gomp_cpu_affinity != NULL)
    gomp_init_thread_affinity (attr);

      err = pthread_create (&pt, attr, gomp_thread_start, start_data);
      if (err != 0)
    gomp_fatal ("Thread creation failed: %s", strerror (err));
    }

下面的程序就是最终启动线程的源程序,能够看到这是一个 for 循环并且启动 nthreads 个线程,pthread_create 是真正创立了线程的代码,并且让线程执行函数 gomp_thread_start 能够看到线程不是间接执行 subfunction 而是将这个函数指针保留到 start_data 当中,并且在函数 gomp_thread_start 真正去调用这个函数,看到这里大家应该明确了整个 parallel construct 的整个流程了。

gomp_thread_start 的函数题也绝对比拟长,在这里咱们选中其中的比拟重要的几行代码,其余的代码进行省略。比照下面线程启动的 pthread_create 语句咱们能够晓得,上面的程序真正的调用了 subfunction,并且给这个函数传递了对应的参数。

static void *
gomp_thread_start (void *xdata)
{
  struct gomp_thread_start_data *data = xdata;
  /* Extract what we need from data.  */
  local_fn = data->fn;
  local_data = data->fn_data;
  local_fn (local_data);
  return NULL;
}

GOMP_parallel_end 剖析

这个函数的次要作用就是一个同步点,保障所有的线程都执行实现之后再持续往后执行,这一部分的源代码比拟杂,其外围原理就是应用路障 barrier 去实现的,这其中是 OpenMP 本人实现的一个 barrier 而不是间接应用 pthread 当中的 barrier,这一部分的源程序就不进行仔细分析了,感兴趣的同学能够自行浏览,能够参考 OpenMP 锁实现原理。

总结

在本篇文章当中次要给大家介绍了 parallel construct 的实现原理,以及他的动静库函数的调用以及源代码剖析,大家只须要理解整个流程不太须要死扣细节(这并无很大的用途)只有当咱们本人须要去实现 OpenMP 的时候须要去理解这些细节,不然咱们只须要理解整个动静库的设计原理即可!


更多精彩内容合集可拜访我的项目:https://github.com/Chang-LeHu…

关注公众号:一无是处的钻研僧,理解更多计算机(Java、Python、计算机系统根底、算法与数据结构)常识。

退出移动版