Increased the ratio of GEMM tuning results to explore; reduced the tuning search space to have a better chance to evaluate more likely parameter combinations

1 files changed, 4 insertions, 4 deletions

diff --git a/src/tuning/kernels/xgemm.cpp b/src/tuning/kernels/xgemm.cpp
index 898b8435..eb7c8a66 100644
--- a/src/tuning/kernels/xgemm.cpp
+++ b/src/tuning/kernels/xgemm.cpp
@@ -48,7 +48,7 @@ class TuneXgemm {
   static size_t DefaultM() { return 1024; }
   static size_t DefaultN() { return 1024; }
   static size_t DefaultK() { return 1024; }
-  static double DefaultFraction() { return 2048.0; }
+  static double DefaultFraction() { return 256.0; }
 
   // Describes how to obtain the sizes of the buffers
   static size_t GetSizeX(const Arguments<T> &) { return 1; } // N/A for this kernel
@@ -67,9 +67,9 @@ class TuneXgemm {
     tuner.AddParameter(id, "NDIMC", {8, 16, 32});
     tuner.AddParameter(id, "MDIMA", {8, 16, 32});
     tuner.AddParameter(id, "NDIMB", {8, 16, 32});
-    tuner.AddParameter(id, "KWI", {2, 8});
-    tuner.AddParameter(id, "VWM", {1, 2, 4, 8});
-    tuner.AddParameter(id, "VWN", {1, 2, 4, 8});
+    tuner.AddParameter(id, "KWI", {2});
+    tuner.AddParameter(id, "VWM", {1, 2, 4});
+    tuner.AddParameter(id, "VWN", {1, 2, 4});
     tuner.AddParameter(id, "STRM", {0, 1});
     tuner.AddParameter(id, "STRN", {0, 1});
     tuner.AddParameter(id, "SA", {0, 1});