update align32 strategy for mm_kernel

Author: root

src/flag_gems/ops/mm.py

@@ -16,7 +16,7 @@
 @libtuner(
     configs=runtime.get_tuned_config("mm"),
     key=["M", "N", "K"],
-    strategy=["log", "log", "log"],
+    strategy=["align32", None, None],
 )
 @triton.heuristics(runtime.get_heuristic_config("mm"))
 @triton.jit

src/flag_gems/utils/libentry.py

@@ -66,6 +66,7 @@ def all_kwargs(self):
 STRATEGY = {
     None: lambda v: v,
     "log": lambda v: math.ceil(math.log2(v)),
+    "align32": lambda v: (v // 32) * 32,
 }