diff --git a/loopy/statistics.py b/loopy/statistics.py index dc938ef70f151654745f8d457f7ea584eeceb84d..2df2cb53ccd05ccf93b09e4416b1784224bb45a1 100755 --- a/loopy/statistics.py +++ b/loopy/statistics.py @@ -1045,13 +1045,13 @@ def get_unused_hw_axes_factor(knl, insn, disregard_local_axes, space=None): result = 1 for iaxis, size in enumerate(size): if iaxis not in used_axes: - result = result * size + if not isinstance(size, int): + if space is not None: + size = size.align_params(space) - if not isinstance(result, int): - if space is not None: - result = result.align_params(space) + size = isl.PwQPolynomial.from_pw_aff(size) - result = isl.PwQPolynomial.from_pw_aff(result) + result = result * size return result