PEP8 loopy.kernel.tools.

loopy/kernel/tools.py

@@ -25,14 +25,10 @@ THE SOFTWARE.
 """
 
 
-
-
 import numpy as np
 from islpy import dim_type
 
 
-
-
 # {{{ add and infer argument dtypes
 
 def add_argument_dtypes(knl, dtype_dict):
@@ -65,10 +61,12 @@ def add_argument_dtypes(knl, dtype_dict):
 
     return knl.copy(args=new_args)
 
+
 def get_arguments_with_incomplete_dtype(knl):
     return [arg.name for arg in knl.args
             if arg.dtype is None]
 
+
 def add_and_infer_argument_dtypes(knl, dtype_dict):
     knl = add_argument_dtypes(knl, dtype_dict)
 
@@ -77,6 +75,7 @@ def add_and_infer_argument_dtypes(knl, dtype_dict):
 
 # }}}
 
+
 # {{{ find_all_insn_inames fixed point iteration
 
 def find_all_insn_inames(kernel):
@@ -181,6 +180,7 @@ def find_all_insn_inames(kernel):
 
 # }}}
 
+
 # {{{ set operation cache
 
 class SetOperationCacheManager:
@@ -192,7 +192,7 @@ class SetOperationCacheManager:
         hashval = hash(set)
         bucket = self.cache.setdefault(hashval, [])
 
-        for bkt_set, bkt_op, bkt_args, result  in bucket:
+        for bkt_set, bkt_op, bkt_args, result in bucket:
             if set.plain_is_equal(bkt_set) and op == bkt_op and args == bkt_args:
                 return result
 
@@ -233,6 +233,7 @@ class SetOperationCacheManager:
 
 # }}}
 
+
 # {{{ domain change helper
 
 class DomainChanger:
@@ -268,6 +269,7 @@ class DomainChanger:
 
 # }}}
 
+
 # {{{ graphviz / dot export
 
 def get_dot_dependency_graph(kernel, iname_cluster=False, iname_edge=True):
@@ -291,6 +293,7 @@ def get_dot_dependency_graph(kernel, iname_cluster=False, iname_edge=True):
 
 # }}}
 
+
 # {{{ domain parameter finder
 
 class DomainParameterFinder:
@@ -323,8 +326,8 @@ class DomainParameterFinder:
                             try:
                                 # friggin' overkill :)
                                 param_expr = solve_affine_equations_for(
-                                        [dep.name], [(shape_i, var("shape_i"))]) \
-                                                [dep.name]
+                                        [dep.name], [(shape_i, var("shape_i"))]
+                                        )[dep.name]
                             except:
                                 # went wrong? oh well
                                 pass