diff options
| author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-10-13 17:40:32 +0100 |
|---|---|---|
| committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-10-31 10:16:25 +0000 |
| commit | 29254aeb11a76c86449c2f38587e9144b2f2aacb (patch) | |
| tree | ca2df26e81c2417b34768ac325e0f7200b5265df /SConscript | |
| parent | e5362e7e5dbccf81c5296a7e77154e11e1a14d2f (diff) | |
| download | ComputeLibrary-29254aeb11a76c86449c2f38587e9144b2f2aacb.tar.gz | |
Optimize CL softmax
* The new softmax implementation consists of only a single kernel.
- There are 2 versions of softmax, one for the x dimension
and one for any other dimensions.
- Softmax kernel handles both native and quantized data type.
Resolves: COMPMID-6447
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I4a9ae5bc63f78aebeaa85ee48a0d102c9c245eda
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10489
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'SConscript')
| -rw-r--r-- | SConscript | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/SConscript b/SConscript index fab9b65acc..8685d01388 100644 --- a/SConscript +++ b/SConscript @@ -451,7 +451,6 @@ if env['opencl'] and env['embed_kernels']: 'src/core/CL/cl_kernels/common/select.cl', 'src/core/CL/cl_kernels/common/slice_ops.cl', 'src/core/CL/cl_kernels/common/softmax_layer.cl', - 'src/core/CL/cl_kernels/common/softmax_layer_quantized.cl', 'src/core/CL/cl_kernels/common/stack_layer.cl', 'src/core/CL/cl_kernels/common/tile.cl', 'src/core/CL/cl_kernels/common/transpose.cl', |