MCPcopy Create free account
hub / github.com/MegEngine/MegCC / MatrixMulForward>

Method MatrixMulForward>

compiler/test/kernel/common/src/cc_fill_attr.cpp:305–317  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

303
304template <>
305KernelGenRet opr_fill_attr<megdnn::MatrixMulForward>(
306 std::unordered_map<std::string, CCAttr>& attr_map,
307 megdnn::MatrixMulForward* opr, const TensorNDArray& tensors,
308 KernelGen::Arch arch,
309 const std::unordered_map<std::string, CCAttr>& proxy_attr) {
310 auto param = opr->param();
311 FILL_MAP(attr_map, param, transposeA);
312 FILL_MAP(attr_map, param, transposeB);
313
314 FILL_MAP_EX(attr_map, param, format, dnnparam_2_str);
315 FILL_MAP_EX(attr_map, param, compute_mode, dnnparam_2_str);
316 return KernelGen::KernelPack::GetKernel(KernType::MatrixMulKernel, arch);
317}
318
319template <>
320KernelGenRet opr_fill_attr<megdnn::BatchedMatrixMulForward>(

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected