Rearranging loops for efficiency

MintonGroup · Jul 23, 2021 · 8070dc5 · 8070dc5
1 parent e47a6f1
commit 8070dc5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/obl/obl.f90 b/src/obl/obl.f90
@@ -57,8 +57,8 @@ module subroutine obl_acc_pl(self, system)
             cb%aobl(i) = -sum(pl%Gmass(1:npl) * pl%aobl(i, 1:npl)) / cb%Gmass
          end do
 
-         do i = 1, NDIM
-            pl%ah(i, 1:npl) = pl%ah(i, 1:npl) + pl%aobl(i, 1:npl) - cb%aobl(i)
+         do i = 1, npl
+            pl%ah(:, i) = pl%ah(:, i) + pl%aobl(:, i) - cb%aobl(:)
          end do
       end associate
 
@@ -89,8 +89,8 @@ module subroutine obl_acc_tp(self, system)
             aoblcb = cb%aoblend
          end if
 
-         do i = 1, NDIM
-            tp%ah(i, 1:ntp) = tp%ah(i, 1:ntp) + tp%aobl(i, 1:ntp) - aoblcb(i)
+         do i = 1, ntp
+            tp%ah(:, i) = tp%ah(:, i) + tp%aobl(:, i) - aoblcb(:)
          end do
 
       end associate