diff options
| author | Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> | 2023-12-05 14:27:31 +0000 |
|---|---|---|
| committer | Mohmun02 <MohammedSuhail.Munshi@arm.com> | 2024-01-10 09:56:39 +0000 |
| commit | 7467ba8fac0afb19d750b3bdda9ba95002634038 (patch) | |
| tree | e47d4989251f03d13590e6b22d9bd228fd1efe34 /src/BUILD.bazel | |
| parent | 7fe7791468978429ab02343a8485b51b39832027 (diff) | |
| download | ComputeLibrary-7467ba8fac0afb19d750b3bdda9ba95002634038.tar.gz | |
Use look up table for fp16 activation
- Enables FP16 lut for logistic activation
- Adds LUTManager to re-use lut where appropriate.
Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com>
Change-Id: I94667b63b452a8e58a1eb59cb0b5866178954523
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10864
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/BUILD.bazel')
| -rw-r--r-- | src/BUILD.bazel | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/BUILD.bazel b/src/BUILD.bazel index f0c4b52688..9d5ae63484 100644 --- a/src/BUILD.bazel +++ b/src/BUILD.bazel @@ -335,6 +335,7 @@ filegroup( "cpu/kernels/elementwise_unary/generic/sve/fp32.cpp", "cpu/kernels/elementwise_unary/generic/sve/impl.cpp", "cpu/kernels/elementwise_unary/generic/sve/integer.cpp", + "cpu/kernels/lut/generic/sve/u16.cpp", "cpu/kernels/scale/sve/fp16.cpp", "cpu/kernels/scale/sve/fp32.cpp", "cpu/kernels/scale/sve/integer.cpp", @@ -637,6 +638,7 @@ filegroup( "core/Utils.cpp", "core/Validate.cpp", "core/Version.cpp", + "core/helpers/LUTManager.cpp", "core/helpers/SoftmaxHelpers.cpp", "core/helpers/Utils.cpp", "core/helpers/WindowHelpers.cpp", |