Skip to content

Commit fde18d7

Browse files
committed
compiler: Improve CIRE's cost model
1 parent a97516f commit fde18d7

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

devito/passes/clusters/aliases.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1011,6 +1011,11 @@ def pick_best(variants):
10111011
indexeds1 = search(i.exprs, Indexed)
10121012
functions1.update({i.function for i in indexeds1})
10131013

1014+
# Filter out objects that are extremely likely to be in cache if not
1015+
# in registers
1016+
functions0 = {f for f in functions0 if f.ndim >= 2}
1017+
functions1 = {f for f in functions1 if f.ndim >= 2}
1018+
10141019
nfunctions0 = len(functions0)
10151020
nfunctions1 = len(functions1)
10161021

0 commit comments

Comments
 (0)