Skip to content
Snippets Groups Projects
Commit 90d98074 authored by Nicholas Christensen's avatar Nicholas Christensen
Browse files

Appease Flake8

parent 7819362f
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 3 additions and 2 deletions
grudge/execution.py
+ 3
− 2
View file @ 90d98074
...@@ -529,7 +529,8 @@ class ExecutionMapper(mappers.Evaluator, ...@@ -529,7 +529,8 @@ class ExecutionMapper(mappers.Evaluator,
continue continue
# Cache operator # Cache operator
cache_key = "diff_batch", in_grp, out_grp, tuple(insn.operators), field.dtype cache_key = "diff_batch", in_grp, out_grp, tuple(insn.operators),\
field.dtype
try: try:
matrices_ary_dev = self.bound_op.operator_data_cache[cache_key] matrices_ary_dev = self.bound_op.operator_data_cache[cache_key]
except KeyError: except KeyError:
...@@ -540,7 +541,7 @@ class ExecutionMapper(mappers.Evaluator, ...@@ -540,7 +541,7 @@ class ExecutionMapper(mappers.Evaluator,
matrices_ary[i] = matrices[op.rst_axis] matrices_ary[i] = matrices[op.rst_axis]
matrices_ary_dev = self.array_context.from_numpy(matrices_ary) matrices_ary_dev = self.array_context.from_numpy(matrices_ary)
self.bound_op.operator_data_cache[cache_key] = matrices_ary_dev self.bound_op.operator_data_cache[cache_key] = matrices_ary_dev
self.array_context.call_loopy( self.array_context.call_loopy(
prg(noperators), prg(noperators),
diff_mat=matrices_ary_dev, diff_mat=matrices_ary_dev,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment