使用 OpenMP 在 Matlab 的 Mex 函数 (Fortran) 中并行化循环

问题描述

我在 Matlab 的 Mex 函数（用 Fortran 编写）中有一个 do 循环，它对 FEM 网格的每个元素执行一些计算。我的网格由 250k 个元素组成，所以我认为值得对其进行并行化。这是我第一次尝试将此代码与 OpenMP 并行化（我是编码初学者）。我使用 reduction 命令来避免 fintk(dofele) = fintk(dofele) + fintele 中的竞争条件。这是对的吗？我可以在 Matlab 中编译它，没有任何问题。但是，当我使用它（在 Matlab 中）时，它会为 12k 元素网格生成正确的结果，并且比序列化的更快，但是当我尝试将它用于 250k 元素网格时，Matlab 崩溃了。谢谢你帮我

  subroutine loop_over_elements( &
  ! OUT  
  fintk,Sxyz,&
  ! IN                    
  Elem,Bemesh,Dofelemat,u,dt,NE,NDOF)

  use omp_lib
  
  implicit none
  
  mwSize NE,NDOF,ele
  integer,parameter   :: dp = selected_real_kind(15,307)
  real(dp) :: fintk(NDOF),Sxyz(6,NE),Elemat(4,Bemesh(6,12,Dofelemat(12,NE)
  real(dp) :: u(NDOF)  
  
  real(dp) :: Bele(6,12),fintele(12),uele(12),si(6),dt
  integer*4  :: nodes(4),dofele(12) 
  
  fintk = 0.D0
  
  !$OMP PARALLEL DO REDUCTION(+:fintk(:)) PRIVATE(ele,nodes,Bele,dofele,uele,si,fintele) 
      DO ele = 1,NE 
        nodes   = Elemat(1:4,ele)
        Bele    = Bemesh(1:6,1:12,ele)
        dofele  = Dofelemat(1:12,ele)
        uele    = u(dofele)
        
        call comput_subroutine( &
!       IN
        Bele,&
!       OUT                                        
        si) 
        
        Sxyz(:,ele)   = si
        fintele       = MATMUL(TRANSPOSE(Bele),si)
        fintk(dofele) = fintk(dofele) + fintele
        
      END DO

  !$OMP END PARALLEL DO

  return
  end

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）