diff options
Diffstat (limited to 'src/core/CL/cl_kernels/batchnormalization_layer.cl')
-rw-r--r-- | src/core/CL/cl_kernels/batchnormalization_layer.cl | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/core/CL/cl_kernels/batchnormalization_layer.cl b/src/core/CL/cl_kernels/batchnormalization_layer.cl index 60307bc9a7..0bb6cd7c5f 100644 --- a/src/core/CL/cl_kernels/batchnormalization_layer.cl +++ b/src/core/CL/cl_kernels/batchnormalization_layer.cl @@ -32,7 +32,9 @@ #if defined(VEC_SIZE) && defined(DATA_TYPE) #if defined(FUSED_ACTIVATION) -#include "activation_layer.cl" +#define SELECT_TYPE VEC_DATA_TYPE(SELECT_DATA_TYPE, VEC_SIZE) +#define TYPE VEC_DATA_TYPE(DATA_TYPE, VEC_SIZE) +#include "activation_helpers.h" #define ACTIVATION_FUNC(x) ACTIVATION_OP(FUSED_ACTIVATION, x) #else /* defined(FUSED_ACTIVATION) */ #define ACTIVATION_FUNC(x) (x) |