Skip to content

Commit

Permalink
yask: Fixup noop operators
Browse files Browse the repository at this point in the history
  • Loading branch information
FabioLuporini committed Jan 8, 2020
1 parent 0ca36b2 commit fec5d95
Showing 1 changed file with 17 additions and 12 deletions.
29 changes: 17 additions & 12 deletions devito/yask/operator.py
Expand Up @@ -113,18 +113,6 @@ def key(i):
super(YASKOmpizer, self).__init__(key=key)


class YASKNoopOperator(Operator):

@classmethod
def _specialize_iet(cls, graph, **kwargs):
# Symbol definitions
data_manager = DataManager()
data_manager.place_definitions(graph)
data_manager.place_casts(graph)

return graph


class YASKOperator(Operator):

"""
Expand Down Expand Up @@ -281,6 +269,23 @@ def __setstate__(self, state):
self.yk_solns[(dimensions, yk_soln_obj)] = yk_soln


class YASKNoopOperator(YASKOperator):

@classmethod
def _specialize_iet(cls, graph, **kwargs):
yk_solns = kwargs['yk_solns']

# Create YASK kernels
make_yask_kernels(graph, yk_solns=yk_solns)

# Symbol definitions
data_manager = DataManager()
data_manager.place_definitions(graph)
data_manager.place_casts(graph)

return graph


class YASKCustomOperator(YASKOperator):

@classmethod
Expand Down

0 comments on commit fec5d95

Please sign in to comment.