[Good First Issue] [ARM]: Implement CPU plugin just-in-time emitter for LeakyRelu operation

[a-sidorova]

Context

JIT Emitters are part of code generation feature (a.k.a. tensor compiler) that automatically produces highly-efficient optimized fused subgraph binary code. Each emitter implements specific operation from low level OpenVINO dialect.

Prerequisites

Recommended to use ARM CPU based platform for development (e.g. Mac, Raspberry Pi etc). Emulators (e.g. QEMU) is still an option, but not that convenient, especially for final performance evaluation.

What needs to be done?

• CPU Plugin has optimization for PRelu operation with scalar second input slope - converts such PRelu ops to CPU-opset op LeakyRelu. Currently, CPU Plugin has JIT Emitters only for PRelu op with non-scalar second input and Relu op with zero alpha for ARM64 SIMD platforms. Need to update existing JIT emitter for Relu to support LeakyRelu op - PRelu op with scalar non-zero alpha value. Use OpenVINO CPU plugin JIT emitters documentation and Arm A-profile A64 Instruction Set Architecture documentation for details.
• Remove limitation condition from ARM64 JIT Executor.

Before emitter implementation, please, modify tests to be sure that developed functionality is covered by test:

• Add jit kernel check in activation test, the test must fail.
• If test doesn't fail then add supported activation type LeakyRelu in CPU activation test instantiation if it's absent.

Tests

Tests are disabled in default build, so ensure to add -DENABLE_TESTS=ON into cmake command.

GoogleTest is used for testing. CPU functional test target is ov_cpu_func_tests. You can use GoogleTest filter:

./bin/[platform]/[build_type]/ov_cpu_func_tests --gtest_filter="*smoke*Activation*LeakyRelu*"

Example Pull Requests

• Abs operation: [CPU] [ARM64] jit abs #23692
• Sigmoid operation, which reuses Exp emitter: [CPU] [ARM64] jit sigmoid #23065

Resources

• Contribution guide - start here!
• What is OpenVINO?
• CPU plugin JIT emitters
• Blog post on contributing to OpenVINO
• User documentation
• Intel DevHub Discord channel - engage in discussions, ask questions and talk to OpenVINO developers
• How to link your Pull Request to an issue

Contact points

@a-sidorova, @dmitry-gorokhov

[Anirudh77715]

Hi Can you please assign this issue to me

[a-sidorova]

@Anirudh77715 hola! Assigned it to you. Good luck! If you have any questions, feel free to ask them 😊