| 4 | using namespace megcc::KernelGen; |
| 5 | |
| 6 | TEST(GI, Fp32MatMulM4N12) { |
| 7 | Checker<MatrixMulForward> checker(Arch::BAREMETAL); |
| 8 | MatrixMulForward::Param param; |
| 9 | checker.set_epsilon(2e-4); |
| 10 | checker.set_kernel_symbol("GI_kernel_fp32_matmul_4x12_.*"); |
| 11 | for (bool trans_a : {false, true}) |
| 12 | for (bool trans_b : {true, false}) |
| 13 | for (size_t m : {1, 3, 58, 9, 10, 12, 23, 67}) |
| 14 | for (size_t n : {1, 5, 6, 7, 8, 9, 10, 12, 24, 33}) |
| 15 | for (size_t k : {1, 3, 5}) { |
| 16 | size_t a0 = m; |
| 17 | size_t a1 = k; |
| 18 | size_t b0 = k; |
| 19 | size_t b1 = n; |
| 20 | if (trans_a) { |
| 21 | a0 = k, a1 = m; |
| 22 | } |
| 23 | if (trans_b) { |
| 24 | b0 = n, b1 = k; |
| 25 | } |
| 26 | param.transposeA = trans_a; |
| 27 | param.transposeB = trans_b; |
| 28 | checker.set_param(param); |
| 29 | checker.execs({{a0, a1}, {b0, b1}, {}}); |
| 30 | } |
| 31 | } |
| 32 | |
| 33 | TEST(GI, Fp32MatMulM4N8K4) { |
| 34 | Checker<MatrixMulForward> checker(Arch::BAREMETAL); |