MCPcopy Create free account
hub / github.com/MegEngine/MegCC / TEST

Function TEST

compiler/test/kernel/opr/naive/fp32Gevm.cpp:5–30  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

Callers

nothing calls this directly

Calls 1

execsMethod · 0.45

Tested by

no test coverage detected