diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-11 17:49:10 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | ac314c25f41e3b2be2ef9073377079584fc88861 (patch) | |
| tree | a654217d60a961288f0a0b3fccc7e5714a2f05bc /src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h | |
| parent | cbf39c63a6eb89a2c80b2338afc374081803d79d (diff) | |
| download | ComputeLibrary-ac314c25f41e3b2be2ef9073377079584fc88861.tar.gz | |
COMPMID-1563: Fix name of NEGEMMInterleavedWrapper
Change-Id: I5f868091cae7bd86eeeb7216d44f32c190c5a604
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/147804
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h')
| -rw-r--r-- | src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h | 24 |
1 files changed, 17 insertions, 7 deletions
diff --git a/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h b/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h index 26a8ade461..00e483c0f8 100644 --- a/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h +++ b/src/core/NEON/kernels/assembly/NEGEMMInterleavedStrategies.h @@ -47,49 +47,59 @@ struct Kernel { }; +#define DEFINE_STRATEGY_SUFFIX(strat, suffix) \ + using strategy = arm_gemm::strat; \ + static constexpr const char *name = #strat suffix; + +#define DEFINE_STRATEGY(strat) \ + DEFINE_STRATEGY_SUFFIX(strat, "") + #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC template <> struct Kernel<float16_t, false> { - using strategy = arm_gemm::hgemm_24x8; + DEFINE_STRATEGY(hgemm_24x8) }; #endif /*__ARM_FEATURE_FP16_VECTOR_ARITHMETIC */ #ifdef __aarch64__ template <> struct Kernel<float, false> { - using strategy = arm_gemm::sgemm_12x8; + DEFINE_STRATEGY(sgemm_12x8) }; template <> struct Kernel<int8_t, false> { - using strategy = arm_gemm::gemm_s8_4x4; + DEFINE_STRATEGY(gemm_s8_4x4) }; template <> struct Kernel<uint8_t, false> { - using strategy = arm_gemm::gemm_u8_4x4; + DEFINE_STRATEGY(gemm_u8_4x4) }; //Use different strategies for 8bit dot product: template <> struct Kernel<int8_t, true> { - using strategy = arm_gemm::gemm_s8_12x8; + DEFINE_STRATEGY_SUFFIX(gemm_s8_12x8, "_dot") }; template <> struct Kernel<uint8_t, true> { - using strategy = arm_gemm::gemm_u8_12x8; + DEFINE_STRATEGY_SUFFIX(gemm_u8_12x8, "_dot") }; #else template <> struct Kernel<float, false> { - using strategy = arm_gemm::sgemm_8x6; + DEFINE_STRATEGY(sgemm_8x6) }; #endif /* __aarch64__ */ +#undef DEFINE_STRATEGY +#undef DEFINE_STRATEGY_SUFFIX + } // namespace } // namespace arm_compute #endif /* __ARM_COMPUTE_NEGEMMINTERLEAVEDSTRATEGIES_H__ */ |