mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-17 15:20:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dnn(ocl): fix fp16 kernel compilation

Browse Source
This commit is contained in:
Alexander Alekhin 2021-12-28 18:02:52 +00:00
parent 92925e846d
commit 76fb3652fc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/dnn/src/opencl/activations.cl
View File

@ -278,7 +278,7 @@ __kernel void CeluForward(const int n, __global T* in, __global T* out,
{ {
int index = get_global_id(0); int index = get_global_id(0);
if(index < n) if(index < n)
out[index] = max(0.f, in[index]) + min(0.f, alpha * expm1(in[index] / alpha)); out[index] = max((T)0.f, in[index]) + (T)min(0.f, alpha * expm1(in[index] / alpha));
} }
__kernel void HardSigmoidForward(const int n, __global T* in, __global T* out, __kernel void HardSigmoidForward(const int n, __global T* in, __global T* out,
@ -287,7 +287,7 @@ __kernel void HardSigmoidForward(const int n, __global T* in, __global T* out,
{ {
int index = get_global_id(0); int index = get_global_id(0);
if(index < n) if(index < n)
out[index] = max(0.f, min(1.f, alpha * in[index] + beta)); out[index] = max((T)0.f, (T)min(1.f, alpha * in[index] + beta));
} }
__kernel void SeluForward(const int n, __global T* in, __global T* out, __kernel void SeluForward(const int n, __global T* in, __global T* out,