MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / kern

Method kern

dnn/src/fallback/matrix_mul/generic_strategy.cpp:30–40  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

28}
29
30void sgemm_8x12::kern(
31 const float* packA, const float* packB, size_t M, size_t N, size_t K, float* C,
32 size_t LDC, bool is_first_k, const float*, float*) const {
33 megdnn_assert(
34 A_dtype.enumv() == B_dtype.enumv() && A_dtype.enumv() == C_dtype.enumv() &&
35 A_dtype.enumv() == DTypeEnum::Float32);
36 MEGDNN_MARK_USED_VAR(A_dtype);
37 MEGDNN_MARK_USED_VAR(B_dtype);
38 MEGDNN_MARK_USED_VAR(C_dtype);
39 gemm_kern(packA, packB, M, N, K, C, LDC, is_first_k, *this);
40}
41
42// vim: syntax=cpp.doxygen

Callers 10

exec_with_ncb_kernMethod · 0.45
get_preprocess_kimplMethod · 0.45
get_kimplMethod · 0.45
executeMethod · 0.45
execute_nakedMethod · 0.45
executeMethod · 0.45
exec_with_ncb_kernMethod · 0.45
dispatch_kernsMethod · 0.45

Calls 2

gemm_kernFunction · 0.70
enumvMethod · 0.45

Tested by

no test coverage detected