From c26228d552f18df9c6251fb35564c66593955ad1 Mon Sep 17 00:00:00 2001 From: mloubout Date: Tue, 10 Oct 2023 10:53:58 -0400 Subject: [PATCH] compiler: prioritize smaller perfect loop over atomics --- devito/passes/iet/parpragma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/devito/passes/iet/parpragma.py b/devito/passes/iet/parpragma.py index 6bfe2a4c423..29ba9a986b3 100644 --- a/devito/passes/iet/parpragma.py +++ b/devito/passes/iet/parpragma.py @@ -289,10 +289,10 @@ def _score_candidate(self, n0, root, collapsable=()): return ( int(n_fp_iters == n), # Fully-parallel nest n_fp_iters_large, + n_fp_iters, n_pia_iters_large, n_pia_iters, -(n0 + 1), # The outer, the better - n_fp_iters, n, )