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

Method usable

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

Source from the content-addressed store, hash-verified

83////////////////////// AlgoF32K8x12x1 ///////////////////////////
84
85bool MatrixMulImpl::AlgoF32K8x12x1::usable(const KernSizeParam& kern_size_param) const {
86 return kern_size_param.compute_mode == param::MatrixMul::ComputeMode::DEFAULT &&
87 kern_size_param.format == param::MatrixMul::Format::DEFAULT &&
88 kern_size_param.B_type == kern_size_param.A_type &&
89 kern_size_param.C_type == kern_size_param.A_type &&
90 kern_size_param.A_type == dtype::Float32{};
91}
92
93size_t MatrixMulImpl::AlgoF32K8x12x1::get_workspace(
94 const KernSizeParam& kern_size_param) const {

Callers 2

get_all_algorithmsMethod · 0.45

Calls 1

enumvMethod · 0.45

Tested by

no test coverage detected