commit 0ff28c945ade28afdf6c2b4b55292bc867c15c25
author Thomas <thomasraoux@google.com> Mon Aug 31 22:39:00 2020 -0700
committer GitHub <noreply@github.com> Mon Aug 31 22:39:00 2020 -0700
tree 43e08975f2334e529f0bab7b29213fbf21cdee60
parent bd9be2381af1b548731837dd3239dd8f91afce35

[ModelBuilder] Fix wrong dimension mapping for second level of tiling. (#3037)

Fix mattmul benchmark, the dimension were mapped backward.

experimental/ModelBuilder/test/BenchMatMulVectorGPU.cpp

diff --git a/experimental/ModelBuilder/test/BenchMatMulVectorGPU.cpp b/experimental/ModelBuilder/test/BenchMatMulVectorGPU.cpp
index 1b5291c..9c8a1f6 100644
--- a/experimental/ModelBuilder/test/BenchMatMulVectorGPU.cpp
+++ b/experimental/ModelBuilder/test/BenchMatMulVectorGPU.cpp
@@ -135,8 +135,8 @@
   Value vSubgroupY = b.create<ConstantIndexOp>(loc, numSubgroupY);
   SmallVector<linalg::ProcInfo, 2> procInfo(2);
   using namespace edsc::op;
-  procInfo[0] = {sg % vSubgroupX, vSubgroupX};
-  procInfo[1] = {sgdiv % vSubgroupY, vSubgroupY};
+  procInfo[0] = {sgdiv % vSubgroupY, vSubgroupY};
+  procInfo[1] = {sg % vSubgroupX, vSubgroupX};
   return procInfo;
 }
 
@@ -325,7 +325,7 @@
               linalg::LinalgTilingOptions()
                   .setLoopType(linalg::LinalgTilingLoopType::ParallelLoops)
                   .setTileSizes(
-                      {tileN / numSubgroupY, tileM / numSubgroupX, tileK})
+                      {tileM / numSubgroupY, tileN / numSubgroupX, tileK})
                   .setDistributionOptions(SGDistribute));
     }
     strategy.vectorize<linalg::MatmulOp>().unrollVector<vector::ContractionOp>(