From 32d5c8899eeb00fcd973fd5e2337ffda9e78073c Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 3 Jun 2013 17:38:40 -0400
Subject: [PATCH] PEP8 loopy.kernel.tools.
---
loopy/kernel/tools.py | 17 ++++++++++-------
1 file changed, 10 insertions(+), 7 deletions(-)
diff --git a/loopy/kernel/tools.py b/loopy/kernel/tools.py
index dadd2c04e..44f3025fd 100644
--- a/loopy/kernel/tools.py
+++ b/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
--
GitLab