diff --git a/loopy/kernel.py b/loopy/kernel.py index 84f7ced462ac6661cdc5f322607fe3e952a31727..04d21a74a120259e4f860504b048d7a227381282 100644 --- a/loopy/kernel.py +++ b/loopy/kernel.py @@ -502,9 +502,7 @@ class LoopKernel(Record): # }}} if assumptions is None: - assumptions_space = domain.get_space() - assumptions_space = assumptions_space.drop_dims( - dim_type.set, 0, assumptions_space.dim(dim_type.set)) + assumptions_space = domain.get_space().params() assumptions = isl.Set.universe(assumptions_space) elif isinstance(assumptions, str):