[codegen/tune]: added fragmentation types

2019-06-06 16:48:32 -07:00
parent f58c9a4d2b
commit cdf5a0d011
7 changed files with 101 additions and 35 deletions
--- a/lib/dnn/gemm.cpp
+++ b/lib/dnn/gemm.cpp
@@ -62,8 +62,8 @@ std::string gemm::src(bool AT, bool BT) {
  }
  std::string res =
 R"(
-const tunable int32 TM = {16, 32, 64, 128};
-const tunable int32 TN = {16, 32, 64, 128};
+const tunable int32 TM = {16};
+const tunable int32 TN = {16};
 const tunable int32 TK = {8};
 const tunable int32 GZ = {1};