diff --git a/loopy/statistics.py b/loopy/statistics.py index 233cfe5e881ef594ebabc536ab8c7b3d18d5cf17..88d7ec328345fd4c97d75b449385316f99c2509d 100755 --- a/loopy/statistics.py +++ b/loopy/statistics.py @@ -1000,6 +1000,9 @@ def add_assumptions_guard(kernel, pwqpolynomial): def count(kernel, set, space=None): try: + if space is not None: + set = set.align_params(space) + return add_assumptions_guard(kernel, set.card()) except AttributeError: pass