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

Method get_all_algorithms_safe

dnn/src/cuda/batched_matrix_mul/opr_impl.cpp:44–49  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

42 return ret;
43}
44std::vector<Algorithm*> BatchedMatrixMulForwardImpl::get_all_algorithms_safe(
45 const TensorLayout& A, const TensorLayout& B, const TensorLayout& C) {
46 auto ret_safe = get_all_algorithms(A, B, C);
47 megdnn_assert(!ret_safe.empty(), "no usable batchedmatrixmulForward fwd algorithm");
48 return ret_safe;
49}
50
51Algorithm* BatchedMatrixMulForwardImpl::get_algorithm_heuristic(
52 const TensorLayout& A, const TensorLayout& B, const TensorLayout& C,

Callers

nothing calls this directly

Calls 2

get_all_algorithmsFunction · 0.85
emptyMethod · 0.45

Tested by

no test coverage detected