xe: gemm: jit: reduce unrolls for minimal [FO]OS

kealan-barbieri · kealan-barbieri · commit db8e8ff73701 · 2025-08-27T16:59:20.000-07:00
diff --git a/src/gpu/intel/jit/gemm/selector/db/kernel.db b/src/gpu/intel/jit/gemm/selector/db/kernel.db
@@ -1101,7 +1101,7 @@ auto _CATALOG_ = kcatalog::toArray({
 {{'F', "gemm", {"T", "T", "S"}, {"T", "N", "N"}}, {5, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {1, 1, 1}, "#AI"}, "at8x2+m16@24 aB16 aB wg 4x2x4 kr cb4x2 ks32 af vav hi pt sr br bk0 sm sn dm grf256 kv afb sys", {16, (LoopType) 255, 256, {(LoopType) 208, (LoopType) 255, (LoopType) 2}, {262144, 262144, 16777216}, {262144, 262144, 32}, {16, 16, 32}, {4, 2, 4}, 1, (WGType) 1, 445, 16384, 16384, {4, 4, 4}, {true, true, true}}, {'E', 17, {1.64426e+06, -145120, -138637, 309031, 2.31014e+06, 2.00704e+06, 0.918257, 1.44548, 0.262092, 1.02772, 0.0377766, 0.0386483, 0.015333, 0.989041, 1.28006, -1.30164, 1.0788e-10}}},
 {{'F', "gemm", {"T", "T", "S"}, {"T", "N", "N"}}, {5, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {1, 1, 1}, "#AI"}, "at16+m16@24 aB16x2 aB wg 2x2x8 kr cb4x2 ks16 xaf vav hi pt sr br bk0 sm sn dm grf256 kv afb sys", {16, (LoopType) 255, 256, {(LoopType) 208, (LoopType) 255, (LoopType) 2}, {262144, 262144, 16777216}, {262144, 262144, 32}, {16, 16, 16}, {2, 2, 8}, 1, (WGType) 1, 445, 8192, 8192, {4, 4, 4}, {true, true, true}}, {'E', 17, {1.65746e+06, -77290.3, -69015.6, 169032, 2.75251e+06, 1.85958e+06, 0.958255, 0.997806, 0.100098, 0.957911, 0.0468295, 0.0427724, 0.0158824, 1, 1.31247, 0.973533, 8.29974e-12}}},
 {{'F', "gemm", {"T", "T", "S"}, {"T", "N", "N"}}, {5, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {1, 1, 1}, "#ABI"}, "at8+m8@16 aB32+m16@16 aB wg 2x2x8 kr af vav hi pt sr br sb32 bk0 sm sn dm grf256 sys", {16, (LoopType) 255, 256, {(LoopType) 208, (LoopType) 255, (LoopType) 2}, {262144, 262144, 16777216}, {262144, 262144, 16777216}, {16, 16, 32}, {2, 2, 8}, 1, (WGType) 1, 261, 0, 4096, {4, 4, 4}, {true, true, true}}, {'E', 17, {1.00059e+06, 567011, 1104.46, 49832.2, 0, 0, 1.24825, 1.37716, 2.24139, 6.36497, 0.0644534, 0.0644534, 0, 0.917445, 1.16564, 0.0446476, 1.86857e-11}}},
-{{'F', "gemm", {"[FO]", "O", "S"}, {"N", "N", "N"}}, {-1, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {8, 8, 1}, "ABIp"}, "am32 am32 aB wg 2x2 sys", {16, (LoopType) 255, 128, {(LoopType) 0, (LoopType) 1, (LoopType) 255}, {8192, 8192, 2048}, {8192, 8192, 2048}, {32, 16, 32}, {2, 2, 1}, 1, (WGType) 0, 257, 0, 0, {8, 8, 4}, {true, true, true}}, {'W', 1, {512}}},
+{{'F', "gemm", {"[FO]", "O", "S"}, {"N", "N", "N"}}, {-1, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {8, 8, 1}, "ABIp"}, "am32 am32 aB wg 2x2 sys", {16, (LoopType) 255, 128, {(LoopType) 0, (LoopType) 1, (LoopType) 255}, {8192, 8192, 2048}, {8192, 8192, 2048}, {16, 16, 32}, {2, 2, 1}, 1, (WGType) 0, 257, 0, 0, {8, 8, 4}, {true, true, true}}, {'W', 1, {512}}},
 {{'F', "gemm", {"[FO]", "O", "S"}, {"N", "N", "N"}}, {-1, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {1, 1, 1}, "#Ip"}, "aB32+m64@32 aB64+m64@32 aB wg 2x8x2 kr cab3 ks64 af hi pt sr br bk0 sn dm grf256 sys l4", {16, (LoopType) 255, 256, {(LoopType) 208, (LoopType) 255, (LoopType) 2}, {524288, 131072, 16777216}, {524288, 131072, 16777216}, {32, 8, 64}, {2, 8, 2}, 1, (WGType) 1, 261, 24576, 24576, {1, 1, 4}, {true, true, true}}, {'E', 17, {1.03799e+06, 666762, 2486.11, 229962, 0, 0, 0.52045, 0.670568, 0.707588, 2.31307, 0.00907287, 0.00907287, 0, 1, 1.37325, 0.975844, 1.68113e-12}}},
 {{'F', "gemm", {"[FO]", "O", "S"}, {"N", "N", "N"}}, {-1, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {8, 8, 1}, "ABIp"}, "av64 am64x2+m64@128 aB wg 2x16 ca3x2 ks128 af hi pt sr br bk0 sn grf256 sys", {16, (LoopType) 255, 256, {(LoopType) 208, (LoopType) 255, (LoopType) 255}, {524288, 32768, 16777216}, {524288, 32768, 16777216}, {32, 2, 128}, {2, 16, 1}, 1, (WGType) 1, 257, 24576, 0, {8, 8, 4}, {true, true, true}}, {'E', 17, {991020, 415137, 0, 0, 0, 0, 0.494061, 1.1783, 1.09356, 3.42418, 0.0192037, 0.0192037, 0, 0.99432, 1.16294, -0.903393, 1.39731e-11}}},
 {{'F', "gemm", {"[FO]", "O", "S"}, {"N", "T", "N"}}, {-1, -1, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {-1, -1, -1}, {8, 8, 1}, "ABIp"}, "am32 at32 aB wg 2x2 grf256 sys", {16, (LoopType) 255, 256, {(LoopType) 0, (LoopType) 1, (LoopType) 255}, {8192, 8192, 2048}, {8192, 8192, 2048}, {32, 16, 32}, {2, 2, 1}, 1, (WGType) 0, 257, 0, 0, {8, 8, 4}, {true, true, true}}, {'W', 1, {512}}},
