add mtile for epilogue in gemv

intel · Apr 30, 2024 · ea84cdb · ea84cdb
1 parent 7bb4912
commit ea84cdb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bestla/bestla/bestla_wrapper.h b/bestla/bestla/bestla_wrapper.h
@@ -214,8 +214,8 @@ class LauncherBase {
                    (_config.size[1] - in) * sizeof(CType));
           }
         }
-        Epilogue::forward(_param.paramC.C + _config.loc[1], 0, 0, _config.loc[1], 1, _config.size[1], _param.paramC,
-                          StackTmp, TmpSize);
+        Epilogue::forward(_param.paramC.C + _config.loc[1], _param.paramC.ldc, 0, _config.loc[1], MTILE,
+                          _config.size[1], _param.paramC, StackTmp, TmpSize);
       }
     }
 
@@ -477,8 +477,8 @@ class LauncherIntKBlock {
                    (_config.size[1] - in) * sizeof(CType));
           }
         }
-        Epilogue::forward(_param.paramC.C + _config.loc[1], 0, 0, _config.loc[1], 1, _config.size[1], _param.paramC,
-                          StackTmp, TmpSize);
+        Epilogue::forward(_param.paramC.C + _config.loc[1], _param.paramC.ldc, 0, _config.loc[1], MTILE, _config.size[1],
+                          _param.paramC, StackTmp, TmpSize);
       }
     }