从main调用的函数中的MPI代码并行化

Question

我有一个C语言的串行代码，我想使用MPI进行并行化， 但是我想并行化不在main中的代码区域， 它在另一个功能中。

例如：

main(){
  int a
  int b
  int c

  while (a>0){
    function_a()
  }
}

function_a(){
  int d,e,f,g     
  function_b(d, e)
  function_b(f, g)

}

然后，我想做的是

 function_a(){
  int d,e,f,g
  if (myrank == 0){
     function_b(d, e)
  }else if (myrank == 1) {
     function_b(f, g)
  }
 }

我已经完成了一些代码，例如对大型数组中的元素求和，但实际上我是MPI的初学者。 请欢迎任何建议