diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp')
| -rw-r--r-- | src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp index 07c4769479..45ba8dc423 100644 --- a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp +++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp @@ -69,7 +69,7 @@ public: template<typename T> static PerformanceParameters get_performance_parameters(const CPUInfo *ci) { - if (std::is_same<T, int32_t>::value) { + if (std::is_same<T, uint32_t>::value) { switch (ci->get_cpu_model()) { case CPUModel::A510: return { 2.64, 2.72, 2.64 }; @@ -79,7 +79,7 @@ public: } } - if (std::is_same<T, int8_t>::value) { + if (std::is_same<T, uint8_t>::value) { switch(ci->get_cpu_model()) { case CPUModel::A510: return { 2.64, 1.79, 0.10 }; |