| 303 | |
| 304 | template <> |
| 305 | KernelGenRet 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 | |
| 319 | template <> |
| 320 | KernelGenRet opr_fill_attr<megdnn::BatchedMatrixMulForward>( |
nothing calls this directly
no outgoing calls
no test coverage detected