Browse Source

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

GitOrigin-RevId: bd152428e6
tags/v1.6.0-rc1
Megvii Engine Team 3 years ago
parent
commit
127870a926
1 changed files with 1 additions and 3 deletions
  1. +1
    -3
      dnn/src/common/algo_base.h

+ 1
- 3
dnn/src/common/algo_base.h View File

@@ -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) {


Loading…
Cancel
Save