diff options
author | Roman Gareev <gareevroman@gmail.com> | 2016-06-22 12:11:30 +0000 |
---|---|---|
committer | Roman Gareev <gareevroman@gmail.com> | 2016-06-22 12:11:30 +0000 |
commit | 397a34a08db195717d0357a6e55437ee4f84fd2c (patch) | |
tree | fc9a878f71eec3f4def9a892b8b31209d3bccd9f /polly | |
parent | 5b2503fb3e292cb4e4d41ea1135df440cde9e097 (diff) | |
download | bcm5719-llvm-397a34a08db195717d0357a6e55437ee4f84fd2c.tar.gz bcm5719-llvm-397a34a08db195717d0357a6e55437ee4f84fd2c.zip |
[NFC] Use isl_schedule_node_band_n_member to get the number of dimensions of a band node.
llvm-svn: 273400
Diffstat (limited to 'polly')
-rw-r--r-- | polly/lib/Transform/ScheduleOptimizer.cpp | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/polly/lib/Transform/ScheduleOptimizer.cpp b/polly/lib/Transform/ScheduleOptimizer.cpp index 6d0450f4c8d..03bf5cd11dc 100644 --- a/polly/lib/Transform/ScheduleOptimizer.cpp +++ b/polly/lib/Transform/ScheduleOptimizer.cpp @@ -514,17 +514,12 @@ bool ScheduleTreeOptimizer::isMatrMultPattern( __isl_keep isl_schedule_node *Node) { auto *PartialSchedule = isl_schedule_node_band_get_partial_schedule_union_map(Node); - if (isl_union_map_n_map(PartialSchedule) != 1) - return false; - auto *NewPartialSchedule = isl_map_from_union_map(PartialSchedule); - auto DimNum = isl_map_dim(NewPartialSchedule, isl_dim_in); - if (DimNum != 3) { - isl_map_free(NewPartialSchedule); + if (isl_schedule_node_band_n_member(Node) != 3 || + isl_union_map_n_map(PartialSchedule) != 1) { + isl_union_map_free(PartialSchedule); return false; } - assert(isl_map_dim(NewPartialSchedule, isl_dim_out) == 3 && - "Each schedule dimension should be represented by a union piecewise" - "quasi-affine expression."); + auto *NewPartialSchedule = isl_map_from_union_map(PartialSchedule); NewPartialSchedule = circularShiftOutputDims(NewPartialSchedule); if (containsMatrMult(NewPartialSchedule)) { isl_map_free(NewPartialSchedule); |