Skip to content

Commit e3ad941

Browse files
committed
misc: Faster d_from_access_gen
1 parent 7a9fc1d commit e3ad941

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

devito/ir/support/basic.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1183,12 +1183,10 @@ def d_from_access_gen(self, accesses):
11831183
Generate all flow, anti, and output dependences involving any of
11841184
the given TimedAccess objects.
11851185
"""
1186-
accesses = as_tuple(accesses)
1186+
accesses = set(as_tuple(accesses))
11871187
for d in self.d_all_gen():
1188-
for i in accesses:
1189-
if d.source == i or d.sink == i:
1190-
yield d
1191-
break
1188+
if accesses & {d.source, d.sink}:
1189+
yield d
11921190

11931191
@memoized_meth
11941192
def d_from_access(self, accesses):

0 commit comments

Comments
 (0)