commit 9d4d91ca6a0fc5d5752b6b7822b8c6fea7662b8e
author Roman Gareev <gareevroman@gmail.com> Thu Apr 06 17:25:08 2017 +0000
committer Roman Gareev <gareevroman@gmail.com> Thu Apr 06 17:25:08 2017 +0000
tree 9d6625c5e1901c18e5a8221f4633677b1fe3c61f
parent e0d466342b250312dd7569c3630becda99a2c50c

[FIX] Fix ScheduleTreeOptimizer::optimizeMatMulPattern

Use new values of the dimensions during their permutation.

llvm-svn: 299663

polly/lib/Transform/ScheduleOptimizer.cpp

diff --git a/polly/lib/Transform/ScheduleOptimizer.cpp b/polly/lib/Transform/ScheduleOptimizer.cpp
index f0a54e0..f4b082c 100644
--- a/polly/lib/Transform/ScheduleOptimizer.cpp
+++ b/polly/lib/Transform/ScheduleOptimizer.cpp
@@ -1314,7 +1314,7 @@
   int NewJ = MMI.j == DimOutNum - 3 ? MMI.i : MMI.j;
   int NewK = MMI.k == DimOutNum - 3 ? MMI.i : MMI.k;
   Node = permuteBandNodeDimensions(Node, NewJ, DimOutNum - 2);
-  NewK = MMI.k == DimOutNum - 2 ? MMI.j : MMI.k;
+  NewK = NewK == DimOutNum - 2 ? NewJ : NewK;
   Node = permuteBandNodeDimensions(Node, NewK, DimOutNum - 1);
   auto MicroKernelParams = getMicroKernelParams(TTI, MMI);
   auto MacroKernelParams = getMacroKernelParams(MicroKernelParams, MMI);