Added locality spec to do concurrent

MintonGroup · Feb 18, 2024 · a0881ab · a0881ab
1 parent 1d9c7d1
commit a0881ab
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/collision/collision_generate.f90 b/src/collision/collision_generate.f90
@@ -209,7 +209,11 @@ module subroutine collision_generate_merge(self, nbody_system, param, t)
                   fragments%density(1) = fragments%mass(1) / volume
                   fragments%radius(1) = (3._DP * volume / (4._DP * PI))**(THIRD)
                   if (param%lrotation) then
+#ifdef DOCONLOC
+                     do concurrent(i = 1:NDIM) shared(impactors, fragments, L_spin_new)
+#else
                      do concurrent(i = 1:NDIM)
+#endif
                         fragments%Ip(i,1) = sum(impactors%mass(:) * impactors%Ip(i,:)) 
                         L_spin_new(i) = sum(impactors%L_orbit(i,:) + impactors%L_spin(i,:))
                      end do