From e2428a013014feac81af60f65f9b16cc244327aa Mon Sep 17 00:00:00 2001 From: Usama Arif Date: Thu, 9 May 2019 11:03:17 +0100 Subject: COMPMID-2194: Refactor activation function macro in OpenCL. New Macros for activation. This commit contains the new macros for activation. Only the activation_layer utilizes the new macros in this commit. Change-Id: I2fa8567cc876e8cb67a1e876652bc348b7ed23ea Signed-off-by: Usama Arif Reviewed-on: https://review.mlplatform.org/c/1104 Comments-Addressed: Arm Jenkins Reviewed-by: Gian Marco Iodice Tested-by: Arm Jenkins --- src/core/CL/CLKernelLibrary.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/core/CL/CLKernelLibrary.cpp') diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index df60001343..b2d7e23624 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -1036,6 +1036,9 @@ Kernel CLKernelLibrary::create_kernel(const std::string &kernel_name, const Stri concat_str += " -DARM_COMPUTE_DEBUG_ENABLED"; #endif // defined(ARM_COMPUTE_DEBUG_ENABLED) + GPUTarget gpu_arch = get_arch_from_target(get_target_from_device(_device)); + concat_str += " -DGPU_ARCH=" + support::cpp11::to_string( + static_cast::type>(gpu_arch)); if(fp16_supported()) { concat_str += " -DARM_COMPUTE_OPENCL_FP16_ENABLED=1 "; -- cgit v1.2.1