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.