| 11 | BatchedMatrixMulForwardImpl::AlgoPack BatchedMatrixMulForwardImpl::sm_algo_pack; |
| 12 | |
| 13 | std::string BatchedMatrixMulForwardImpl::AlgoBase::SizeArgs::to_string() const { |
| 14 | auto&& param = opr->param(); |
| 15 | size_t m = layout_a.shape[0], n = layout_b.shape[1], |
| 16 | k = layout_a.shape[param.transposeA ? 0 : 1]; |
| 17 | MEGDNN_MARK_USED_VAR(m); |
| 18 | MEGDNN_MARK_USED_VAR(n); |
| 19 | MEGDNN_MARK_USED_VAR(k); |
| 20 | return ssprintf( |
| 21 | "A={%zux%zu},B={%zux%zu},C={%zux%zu},Transpose A=%d,Transpose " |
| 22 | "B=%d,ldA=%zu,ldB=%zu,ldC=%zu", |
| 23 | m, k, k, n, m, n, param.transposeA, param.transposeB, layout_a.stride[0], |
| 24 | layout_b.stride[0], layout_c.stride[0]); |
| 25 | } |
| 26 | |
| 27 | BatchedMatrixMulForwardImpl::AlgoBase::SizeArgs::SizeArgs( |
| 28 | BatchedMatrixMulForwardImpl* o, const TensorLayout& A, const TensorLayout& B, |
no test coverage detected