diff --git a/loopy/isl_helpers.py b/loopy/isl_helpers.py index 84cebabcbf346a499d31602807af65872c6c5aa2..de0c1f0f12fad00b678da76e7f55400b85df4144 100644 --- a/loopy/isl_helpers.py +++ b/loopy/isl_helpers.py @@ -93,7 +93,7 @@ def pw_aff_to_aff(pw_aff): def dump_space(ls): - return " ".join("%s: %d" % (dt, ls.dim(getattr(dim_type, dt))) + return " ".join("%s: %d" % (dt, ls.dim(getattr(dim_type, dt))) for dt in dim_type.names) diff --git a/loopy/kernel.py b/loopy/kernel.py index 9d78eb961bf02545b1158e85d7912a5458ce2ddc..ffbae5657e650b1894a99695899546169c7decb1 100644 --- a/loopy/kernel.py +++ b/loopy/kernel.py @@ -212,7 +212,7 @@ class TemporaryVariable(Record): class Instruction(Record): """ - :ivar id: An (otherwise meaningless) identifier that is unique within + :ivar id: An (otherwise meaningless) identifier that is unique within a :class:`LoopKernel`. :ivar assignee: :ivar expression: @@ -661,7 +661,7 @@ class LoopKernel(Record): from loopy.isl_helpers import static_max_of_pw_aff try: # insist block size is constant - size = static_max_of_pw_aff(size, + size = static_max_of_pw_aff(size, constants_only=isinstance(tag, LocalIndexTag)) except ValueError: pass diff --git a/loopy/schedule.py b/loopy/schedule.py index f7eb17d4a2a59f5acef773cc313dda9332cde6d6..1c4a276527717c2a232722d0929e561047f253ef 100644 --- a/loopy/schedule.py +++ b/loopy/schedule.py @@ -553,7 +553,7 @@ def assign_automatic_axes(kernel, only_axis_0=True): # of a new axis assignment. if only_axis_0: - # If we were only assigining axis 0, then assign all the remaining + # If we were only assigining axis 0, then assign all the remaining # axes next. return assign_automatic_axes(kernel, only_axis_0=False) else: diff --git a/loopy/symbolic.py b/loopy/symbolic.py index 50a2659ae58e5c8bedd27e4da0799a004073e936..4504daa7d21f63749d02e8fbd9c4cd139fe372c0 100644 --- a/loopy/symbolic.py +++ b/loopy/symbolic.py @@ -85,7 +85,7 @@ class DependencyMapper(DependencyMapperBase): # {{{ functions to primitives class FunctionToPrimitiveMapper(IdentityMapper): - """Looks for invocations of a function called 'cse' or 'reduce' and + """Looks for invocations of a function called 'cse' or 'reduce' and turns those into the actual pymbolic primitives used for that. """