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

Method get_workspace

dnn/src/fallback/matrix_mul/algos.cpp:93–108  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

91}
92
93size_t MatrixMulImpl::AlgoF32K8x12x1::get_workspace(
94 const KernSizeParam& kern_size_param) const {
95 MIDOUT_BEGIN(
96 megdnn_fb_matmul_f32_kern,
97 midout_iv("AlgoF32K8x12x1::get_workspace"_hash)) {
98 auto M = kern_size_param.M, N = kern_size_param.N, K = kern_size_param.K;
99 matmul::fallback::sgemm_8x12 strategy(
100 M, N, K, kern_size_param.A_type, kern_size_param.B_type,
101 kern_size_param.C_type);
102 return matmul::GemmInterleaved<matmul::fallback::sgemm_8x12>(
103 M, N, K, kern_size_param.trA, kern_size_param.trB, strategy)
104 .get_workspace_size();
105 }
106 MIDOUT_END();
107 return 0;
108}
109
110MatrixMulImpl::kern_t MatrixMulImpl::AlgoF32K8x12x1::get_kern(
111 const KernSizeParam&) const {

Callers 2

Calls 3

MIDOUT_BEGINFunction · 0.70
get_workspace_sizeMethod · 0.45
enumvMethod · 0.45

Tested by

no test coverage detected