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

Method BatchedMatrixMulForward>

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

Source from the content-addressed store, hash-verified

318
319template <>
320KernelGenRet opr_fill_attr<megdnn::BatchedMatrixMulForward>(
321 std::unordered_map<std::string, CCAttr>& attr_map,
322 megdnn::BatchedMatrixMulForward* opr, const TensorNDArray& tensors,
323 KernelGen::Arch arch,
324 const std::unordered_map<std::string, CCAttr>& proxy_attr) {
325 auto param = opr->param();
326 FILL_MAP(attr_map, param, transposeA);
327 FILL_MAP(attr_map, param, transposeB);
328
329 FILL_MAP_EX(attr_map, param, format, dnnparam_2_str);
330 FILL_MAP_EX(attr_map, param, compute_mode, dnnparam_2_str);
331 return KernelGen::KernelPack::GetKernel(KernType::BatchMatmulKernel, arch);
332}
333
334template <>
335KernelGenRet opr_fill_attr<megdnn::WarpPerspectiveForward>(

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected