Merge pull request #30 from zjing14/add_m_n_cluster_tuning

Added m/n cluster tuning
llvm · Aug 3, 2020 · ff742a1 · ff742a1
2 parents 976a47d + a644cce
commit ff742a1
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/mlir/lib/Dialect/MIOpen/Transforms/AffixTuningParameters.cpp b/mlir/lib/Dialect/MIOpen/Transforms/AffixTuningParameters.cpp
@@ -171,10 +171,14 @@ void AffixTuningParameters::runOnFunction() {
     // Hard coded parameters, will change in a different pass. Please visit
     // gridwise_convolution_implicit_gemm_v4r4_nchw_kcyx_nkhw for details
     op.setAttr("k_per_thread", b.getI32IntegerAttr(1));
-    op.setAttr("m_level0_cluster", b.getI32IntegerAttr(4));
-    op.setAttr("n_level0_cluster", b.getI32IntegerAttr(4));
-    op.setAttr("m_level1_cluster", b.getI32IntegerAttr(4));
-    op.setAttr("n_level1_cluster", b.getI32IntegerAttr(4));
+    op.setAttr("m_level0_cluster",
+               b.getI32IntegerAttr(blockGemmDerivedParam.gemmMLevel0Cluster));
+    op.setAttr("n_level0_cluster",
+               b.getI32IntegerAttr(blockGemmDerivedParam.gemmNLevel0Cluster));
+    op.setAttr("m_level1_cluster",
+               b.getI32IntegerAttr(blockGemmDerivedParam.gemmMLevel1Cluster));
+    op.setAttr("n_level1_cluster",
+               b.getI32IntegerAttr(blockGemmDerivedParam.gemmNLevel1Cluster));
     op.setAttr("matrix_a_source_vector_read_dim", b.getI32IntegerAttr(0));
     op.setAttr("matrix_b_source_vector_read_dim", b.getI32IntegerAttr(1));
     op.setAttr("matrix_c_source_dest_vector_read_write_dim",