Browse Source

fix(dnn/fallback): fix conv1x1's is_preferred method

GitOrigin-RevId: 6d34080d25
tags/v0.5.0
Megvii Engine Team 5 years ago
parent
commit
1af4863179
4 changed files with 9 additions and 3 deletions
  1. +1
    -1
      dnn/src/aarch64/matrix_mul/opr_impl.cpp
  2. +2
    -2
      dnn/src/armv7/matrix_mul/opr_impl.cpp
  3. +4
    -0
      dnn/src/fallback/conv_bias/conv1x1/algos.h
  4. +2
    -0
      src/core/impl/comp_node/cpu/comp_node.cpp

+ 1
- 1
dnn/src/aarch64/matrix_mul/opr_impl.cpp View File

@@ -64,8 +64,8 @@ public:
all_algos.emplace_back(&int8x8x32_k8x12x4_dotprod);
#else
all_algos.emplace_back(&int8x8x32_gemv);
all_algos.emplace_back(&int8x8x32_k8x8x8);
all_algos.emplace_back(&int8x8x32_k4x4x16);
all_algos.emplace_back(&int8x8x32_k8x8x8);
all_algos.emplace_back(&int8x8x32_mk4_4x4x16);
#endif
all_algos.emplace_back(&int8x8x16_k4x4x16);


+ 2
- 2
dnn/src/armv7/matrix_mul/opr_impl.cpp View File

@@ -61,11 +61,11 @@ public:
all_algos.emplace_back(&int8x8x32_gemv);
#endif
all_algos.emplace_back(&int8x8x32_mk4_4x2x16);
all_algos.emplace_back(&int8x8x32_k4x8x8);
all_algos.emplace_back(&int8x8x32_k4x2x16);
all_algos.emplace_back(&int8x8x32_k4x8x8);
all_algos.emplace_back(&quint8_k4x8x8);
all_algos.emplace_back(&int8x8x16_k4x8x8);
all_algos.emplace_back(&int8x8x16_k4x2x16);
all_algos.emplace_back(&int8x8x16_k4x8x8);
all_algos.emplace_back(&int16x16x32_k12x4x1);
all_algos.emplace_back(&int16x16x32_mk8_4x8);
}


+ 4
- 0
dnn/src/fallback/conv_bias/conv1x1/algos.h View File

@@ -41,6 +41,10 @@ public:
SmallVector<NCBKern> dispatch_kerns(
ConvBiasImpl* opr, const NCBKernSizeParam& param) const override;

bool is_preferred(ConvBiasImpl*, const NCBKernSizeParam&) const override{
return true;
}

protected:
size_t get_oc_tile_size_heuristic(const NCBKernSizeParam& param) const;



+ 2
- 0
src/core/impl/comp_node/cpu/comp_node.cpp View File

@@ -58,7 +58,9 @@ class CpuCompNode::WorkerQueue final
void on_async_queue_worker_thread_start() override {
mgb_assert(m_locator.device >= 0);
if (enable_affinity) {
#if !defined(ANDROID) && !defined(__ANDROID__)
sys::set_cpu_affinity({m_locator.device});
#endif
}
sys::set_thread_name(m_locator.to_string());
if(m_thread_pool)


Loading…
Cancel
Save