diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-07-19 09:54:47 +0100 |
---|---|---|
committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2019-07-23 15:01:41 +0000 |
commit | d1f54767fc9d6398a5eea38e639dd0ce3df8e5d8 (patch) | |
tree | 0e271b739fe9144c22a8cc05852e3fc28db88a7a /tests/CL/Helper.h | |
parent | 5f98d74892468b944246e60c5a70ad84a7c6bbc9 (diff) | |
download | ComputeLibrary-d1f54767fc9d6398a5eea38e639dd0ce3df8e5d8.tar.gz |
COMPMID-1979: Fuse Activation Function in CLGEMM - part 3
Fused beta*bias in in the old cl gemm kernels
Fused activation function in the old cl gemm kernels
Change-Id: I695fb9189e6d4792010abd256784624982d17d79
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1587
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/CL/Helper.h')
-rw-r--r-- | tests/CL/Helper.h | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/CL/Helper.h b/tests/CL/Helper.h index ab2f8ccb22..0a4566be8d 100644 --- a/tests/CL/Helper.h +++ b/tests/CL/Helper.h @@ -53,6 +53,19 @@ public: k->configure(std::forward<Args>(args)...); _kernel = std::move(k); } + /** Configure the kernel setting the GPU target as well + * + * @param[in] gpu_target GPUTarget to set + * @param[in] args Configuration arguments. + */ + template <typename... Args> + void configure(GPUTarget gpu_target, Args &&... args) + { + auto k = arm_compute::support::cpp14::make_unique<K>(); + k->set_target(gpu_target); + k->configure(std::forward<Args>(args)...); + _kernel = std::move(k); + } /** Validate input arguments * * @param[in] args Configuration arguments. |