FlattenAlgo: Ensure we _really_ obtain a param space

This resolves "isl_space.c:1775: not a parameter space" errors I have seen
on two systems.

llvm-svn: 281052

diff --git a/polly/lib/Transform/FlattenAlgo.cpp b/polly/lib/Transform/FlattenAlgo.cpp
index bd1ce4d..03d337d 100644 (file)
--- a/polly/lib/Transform/FlattenAlgo.cpp
+++ b/polly/lib/Transform/FlattenAlgo.cpp
@@ -227,7 +227,8 @@ IslPtr<isl_union_map> tryFlattenSequence(IslPtr<isl_union_map> Schedule) {
   auto ScatterSet =
       give(isl_set_from_union_set(isl_union_map_range(Schedule.copy())));
 
-  auto ParamSpace = give(isl_union_map_get_space(Schedule.keep()));
+  auto ParamSpace =
+      give(isl_space_params(isl_union_map_get_space(Schedule.keep())));
   auto Dims = isl_set_dim(ScatterSet.keep(), isl_dim_set);
   assert(Dims >= 2);