Merge pull request #51 from oneilm/feature/issue-49-openmp-direct

Feature/issue 49 openmp direct
flatironinstitute · Mar 19, 2024 · c83ad43 · c83ad43
2 parents 8f03234 + 96d6497
commit c83ad43
Show file tree

Hide file tree

Showing 3 changed files with 429 additions and 0 deletions.
diff --git a/src/Helmholtz/helmkernels.f b/src/Helmholtz/helmkernels.f
@@ -90,6 +90,8 @@ subroutine h3ddirectcp(nd,zk,sources,charge,ns,ztarg,nt,
 
       zkeye = zk*eye
 
+c$omp parallel do default(shared) 
+c$omp$     private(i, zdiff, j, dd, d, ztmp, idim)
       do i=1,nt
         do j=1,ns
           zdiff(1) = ztarg(1,i)-sources(1,j)
@@ -107,6 +109,7 @@ subroutine h3ddirectcp(nd,zk,sources,charge,ns,ztarg,nt,
  1000     continue
         enddo
       enddo
+c$omp end parallel do      
 
 
       return
@@ -183,6 +186,9 @@ subroutine h3ddirectcg(nd,zk,sources,charge,ns,ztarg,nt,
 
       zkeye = zk*eye
 
+c$omp parallel do default(shared)
+c$omp$    private(i, j, zdiff, dd, d, cd, cd1, ztmp1)
+c$omp$    private(ztmp2, ztmp3, idim)
       do i=1,nt
         do j=1,ns
           zdiff(1) = ztarg(1,i)-sources(1,j)
@@ -206,6 +212,7 @@ subroutine h3ddirectcg(nd,zk,sources,charge,ns,ztarg,nt,
  1000     continue
         enddo
       enddo
+c$omp end parallel do      
 
 
       return
@@ -285,6 +292,8 @@ subroutine h3ddirectdp(nd,zk,sources,
 
       zkeye = zk*eye
 
+c$omp parallel do default(shared)
+c$omp$   private(i, j, zdiff, dd, d, dinv, cd, cd1, idim, dotprod)
       do i=1,nt
         do j=1,ns
           zdiff(1) = ztarg(1,i)-sources(1,j)
@@ -309,6 +318,7 @@ subroutine h3ddirectdp(nd,zk,sources,
  1000     continue
         enddo
       enddo
+c$omp end parallel do      
 
 
       return
@@ -394,6 +404,9 @@ subroutine h3ddirectdg(nd,zk,sources,dipvec,ns,ztarg,nt,pot,
 
       zkeye = zk*eye
 
+c$omp parallel do default(shared)
+c$omp$   private(i, j, zdiff, dd, d, dinv, dinv2, cd, cd2)
+c$omp$   private(cd3, idim, dotprod, cd4)      
       do i=1,nt
         do j=1,ns
           zdiff(1) = ztarg(1,i)-sources(1,j)
@@ -428,6 +441,7 @@ subroutine h3ddirectdg(nd,zk,sources,dipvec,ns,ztarg,nt,pot,
  1000     continue
         enddo
       enddo
+c$omp end parallel do      
 
 
       return
@@ -506,6 +520,8 @@ subroutine h3ddirectcdp(nd,zk,sources,charge,
 
       zkeye = zk*eye
 
+c$omp parallel do default(shared)
+c$omp$    private(i, j, zdiff, dd, d, dinv, cd, cd1, idim, dotprod)
       do i=1,nt
         do j=1,ns
           zdiff(1) = ztarg(1,i)-sources(1,j)
@@ -532,6 +548,7 @@ subroutine h3ddirectcdp(nd,zk,sources,charge,
  1000     continue
         enddo
       enddo
+c$omp end parallel do      
 
 
 
@@ -619,6 +636,9 @@ subroutine h3ddirectcdg(nd,zk,sources,charge,
 
       zkeye = zk*eye
 
+c$omp parallel do default(shared)
+c$omp$   private(i, j, zdiff, dd, d, dinv, dinv2, cd, cd2, cd3)
+c$omp$   private(idim, dotprod, cd4)      
       do i=1,nt
         do j=1,ns
           zdiff(1) = ztarg(1,i)-sources(1,j)
@@ -657,6 +677,7 @@ subroutine h3ddirectcdg(nd,zk,sources,charge,
  1000     continue
         enddo
       enddo
+c$omp end parallel do      
 
 
       return