diff --git a/loopy/statistics.py b/loopy/statistics.py index 7cb70026d7736fa3b5fa2c98fa970a9fac126460..1fe55111c052834d6b610d7f7b5b0d419ef142b6 100755 --- a/loopy/statistics.py +++ b/loopy/statistics.py @@ -937,6 +937,8 @@ class GlobalMemAccessCounter(MemAccessCounter): from loopy.symbolic import CoefficientCollector from loopy.kernel.array import FixedStrideArrayDimTag from pymbolic.primitives import Variable + from loopy.symbolic import simplify_using_aff + from loopy.diagnostic import ExpressionNotAffineError lid_strides = {} for ltag, iname in six.iteritems(lid_to_iname): @@ -944,8 +946,6 @@ class GlobalMemAccessCounter(MemAccessCounter): # check coefficient of this lid for each axis for idx, axis_tag in zip(index, array.dim_tags): - from loopy.symbolic import simplify_using_aff - from loopy.diagnostic import ExpressionNotAffineError try: coeffs = CoefficientCollector()( simplify_using_aff(self.knl, idx)) @@ -980,8 +980,6 @@ class GlobalMemAccessCounter(MemAccessCounter): # check coefficient of this gid for each axis for idx, axis_tag in zip(index, array.dim_tags): - from loopy.symbolic import simplify_using_aff - from loopy.diagnostic import ExpressionNotAffineError try: coeffs = CoefficientCollector()( simplify_using_aff(self.knl, idx))