From d1f54767fc9d6398a5eea38e639dd0ce3df8e5d8 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Fri, 19 Jul 2019 09:54:47 +0100 Subject: 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 Reviewed-on: https://review.mlplatform.org/c/1587 Reviewed-by: Giuseppe Rossini Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- tests/CL/Helper.h | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'tests/CL/Helper.h') 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)...); _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 + void configure(GPUTarget gpu_target, Args &&... args) + { + auto k = arm_compute::support::cpp14::make_unique(); + k->set_target(gpu_target); + k->configure(std::forward(args)...); + _kernel = std::move(k); + } /** Validate input arguments * * @param[in] args Configuration arguments. -- cgit v1.2.1