diff options
| author | Sheri Zhang <sheri.zhang@arm.com> | 2020-04-14 22:29:36 +0100 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-04-15 13:53:05 +0000 |
| commit | 45198c8fe5c262cf7fba6f22cfc03ccf194e8bca (patch) | |
| tree | 8d9c2046a89c703ffe6cc76e55b2b73662fbdc57 /tests/validation/NEON/QLSTMLayerNormalization.cpp | |
| parent | 5fc8f8471e80303230c7664ffb80ddf839150601 (diff) | |
| download | ComputeLibrary-45198c8fe5c262cf7fba6f22cfc03ccf194e8bca.tar.gz | |
COMPMID-3239: Implement QSYMM16 LayerNormalizationKernel for CL
Use NE/CLSynthetizeFunction instead of NE/CLQLSTMLayerNormalizationValidationFixture
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I62ace213a5261f2d307da6953d0521492aa05292
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3019
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/NEON/QLSTMLayerNormalization.cpp')
| -rw-r--r-- | tests/validation/NEON/QLSTMLayerNormalization.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tests/validation/NEON/QLSTMLayerNormalization.cpp b/tests/validation/NEON/QLSTMLayerNormalization.cpp index 248bf5cf78..3d71175a6f 100644 --- a/tests/validation/NEON/QLSTMLayerNormalization.cpp +++ b/tests/validation/NEON/QLSTMLayerNormalization.cpp @@ -26,6 +26,7 @@ #include "arm_compute/runtime/Tensor.h" #include "arm_compute/runtime/TensorAllocator.h" #include "tests/NEON/Accessor.h" +#include "tests/NEON/Helper.h" #include "tests/PaddingCalculator.h" #include "tests/datasets/ShapeDatasets.h" #include "tests/framework/Asserts.h" @@ -46,6 +47,8 @@ namespace constexpr uint32_t vector_size_byte = 16; using test::datasets::ShapeDataset; +using NEQLSTMLayerNormalization = NESynthetizeFunction<NEQLSTMLayerNormalizationKernel>; + template <uint32_t num_elements_per_iter, uint32_t num_batches, uint32_t num_iteration> class QLSTMLayerNormShapeDataSet : public ShapeDataset { @@ -150,7 +153,7 @@ DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, ), input_info, weight_info, bias_info, output_info) { - const Status s = NEQLSTMLayerNormalizationKernel::validate(&input_info, &output_info, &weight_info, &bias_info); + const Status s = NEQLSTMLayerNormalization::validate(&input_info, &output_info, &weight_info, &bias_info); ARM_COMPUTE_EXPECT(!bool(s), framework::LogLevel::ERRORS); } @@ -158,7 +161,7 @@ DATA_TEST_CASE(Validate, framework::DatasetMode::ALL, // *INDENT-ON* template <typename T> -using NEQLSTMLayerNormalizationFixture = NEQLSTMLayerNormalizationValidationFixture<Tensor, Accessor, NEQLSTMLayerNormalizationKernel, T>; +using NEQLSTMLayerNormalizationFixture = QLSTMLayerNormalizationValidationFixture<Tensor, Accessor, NEQLSTMLayerNormalization, T>; TEST_SUITE(Quantized) TEST_SUITE(QSYMM16) |