IntenseWebs/openvino
3
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add condition for activation ceil so it works when data type is fp32 or fp16 only (#16334)

Browse Source
This commit is contained in:
Wilson Seok authored and GitHub committed 2023-03-17 11:46:44 +09:00
1 parent 8d1139b61a
commit c5ccb3e954
1 file changed
+4 -1
Download Patch File Download Diff File Expand all files Collapse all files
src/plugins/intel_gpu/src/kernel_selector/jitter.cpp
+4 -1
View File
@@ -1140,7 +1140,10 @@ JitConstants MakeActivationJitConstants(ActivationFunction activation_function,
jitConstants.AddConstant(MakeJitConstant(macro_def, "(input)"));
break;
case ActivationFunction::CEIL:
jitConstants.AddConstant(MakeJitConstant(macro_def, "(ceil(input))"));
if (out_dt == Datatype::F32 || out_dt == Datatype::F16)
jitConstants.AddConstant(MakeJitConstant(macro_def, "(ceil(input))"));
else
jitConstants.AddConstant(MakeJitConstant(macro_def, "(input)"));
break;
case ActivationFunction::NEGATIVE:
jitConstants.AddConstant(MakeJitConstant(macro_def, "(-input)"));