diff options
Diffstat (limited to 'src/core')
| -rw-r--r-- | src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp b/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp index e3355ed2d5..8bef2b7bae 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp @@ -149,25 +149,25 @@ static const GemmImplementation<float, float> gemm_fp32_methods[] = //Pretranpose, 2D split { GemmMethod::GEMM_INTERLEAVED_2D, - "sgemm_12x8", - [](const GemmArgs &args) { return args._pretransposed_hint; }, + "sgemm_12x8_pretranspose_2d", [](const GemmArgs &args) { return args._pretransposed_hint; }, + [](const GemmArgs &args) { return args._maxthreads >= 8; }, [](const GemmArgs &args) { return new GemmInterleavedPretransposed2d<sgemm_12x8, float, float>(args); } }, //Tranpose, 2D split, no blockmanager { GemmMethod::GEMM_INTERLEAVED_2D, - "sgemm_12x8", - [](const GemmArgs &args) { return (!args._pretransposed_hint) && args._maxthreads >= 8; }, - [](const GemmArgs &args) { return (!args._pretransposed_hint) && args._maxthreads >= 8; }, + "sgemm_12x8_2d", + nullptr, + [](const GemmArgs &args) { return (!args._pretransposed_hint) && (args._maxthreads >= 8); }, [](const GemmArgs &args) { return new GemmInterleaved2d<sgemm_12x8, float, float>(args); } }, //Tranpose, 1D split, with blockmanager { GemmMethod::GEMM_INTERLEAVED, - "sgemm_12x8", - [](const GemmArgs &args) { return (!args._pretransposed_hint); }, - [](const GemmArgs &args) { return (!args._pretransposed_hint); }, + "sgemm_12x8_1d", + nullptr, + nullptr, [](const GemmArgs &args) { return new GemmInterleaved<sgemm_12x8, float, float>(args); } }, |