compiler: remove atomic collapse hack

devitocodes · Sep 18, 2023 · 6ed3785 · 6ed3785
1 parent fa1a9b7
commit 6ed3785
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/devito/passes/iet/parpragma.py b/devito/passes/iet/parpragma.py
@@ -295,9 +295,6 @@ def _select_candidates(self, candidates):
                     except TypeError:
                         pass
 
-                # At least one inner loop (nested) or
-                # we do not collapse most inner loop if it is an atomic reduction
-                if not i.is_ParallelAtomic or nested:
                     collapsable.append(i)
 
             # Give a score to this candidate, based on the number of fully-parallel
@@ -318,9 +315,6 @@ def _select_candidates(self, candidates):
         return root, list(collapsable)
 
     def _make_reductions(self, partree):
-        if not any(i.is_ParallelAtomic for i in partree.collapsed):
-            return partree
-
         exprs = [i for i in FindNodes(Expression).visit(partree) if i.is_reduction]
 
         reductions = []