diff options
| author | SiCongLi <sicong.li@arm.com> | 2021-10-06 15:25:57 +0100 |
|---|---|---|
| committer | SiCong Li <sicong.li@arm.com> | 2021-10-28 11:00:52 +0000 |
| commit | 1af5416917268692fcd4b34b1d7ffebd3a2aea8a (patch) | |
| tree | 81833ecad401eeb0101fb0d464728df8b699caf8 /src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedKernel.h | |
| parent | 49956ccf029ff4c1873e3a6702b5bede95d81f7a (diff) | |
| download | ComputeLibrary-1af5416917268692fcd4b34b1d7ffebd3a2aea8a.tar.gz | |
Add experimental PostOp interface to ClGemmMatrixMultiplyReshapedKernel Part 1
This interface supports the fusion of multiple elementwise operations
Partially resolves: COMPMID-4435
Change-Id: If68dd7dd98dcf239fde7cb1f0a4a6d4d1e899a6f
Signed-off-by: SiCongLi <sicong.li@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6483
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedKernel.h')
| -rw-r--r-- | src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedKernel.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedKernel.h b/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedKernel.h index b320d318e9..09160ec0d1 100644 --- a/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedKernel.h +++ b/src/gpu/cl/kernels/ClGemmMatrixMultiplyReshapedKernel.h @@ -106,6 +106,7 @@ private: bool _add_bias{ false }; bool _export_to_cl_image{ false }; unsigned int _k{ 1 }; + unsigned int _num_post_op_args{ 0 }; // (EXPERIMENTAL_POST_OPS) total number of post op arguments }; } // namespace kernels } // namespace opencl |