提交 127870a9 编写于 作者: M Megvii Engine Team

feat(dnn/opencl): add heuristic rule for batched matmul

GitOrigin-RevId: bd152428e6cb92c34a488dfc2a1992cfd10a6869
上级 d86ed426
......@@ -55,13 +55,11 @@ namespace megdnn {
*/
template <typename AlgoBase>
class AlgoConstructMixin {
private:
std::vector<std::unique_ptr<AlgoBase>> m_refhold;
protected:
std::vector<std::unique_ptr<AlgoBase>> m_refhold;
typename AlgoBase::Mapper m_all_algos_map;
public:
//! construct the algo which described by desc, and return the instance
AlgoBase* construct_and_get_algo(
const detail::Algorithm::Info::Desc& desc) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册