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

lws quantize/eltwise (#11851)

Browse Source
This commit is contained in:
Sungeun Kim authored and GitHub committed 2022-06-13 11:18:49 +09:00
1 parent d831047f30
commit 9bda2bd580
2 files changed
+4 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/plugins/intel_gpu/src/kernel_selector/core/actual_kernels/eltwise/eltwise_kernel_base.cpp
+1
View File
@@ -615,6 +615,7 @@ EltwiseKernelBase::DispatchData EltwiseKernelBase::SetDefault(const eltwise_para
dispatchData.lws[1] = 1;
dispatchData.lws[2] = 32;
} else if ((params.outputs[0].GetLayout() == DataLayout::b_fs_yx_fsv32 ||
params.outputs[0].GetLayout() == DataLayout::b_fs_zyx_fsv32 ||
params.outputs[0].GetLayout() == DataLayout::bs_fs_yx_bsv32_fsv32 ||
params.outputs[0].GetLayout() == DataLayout::bs_fs_zyx_bsv32_fsv32)) {
if (params.layoutBased || params.int8_quantization || params.broadcast) {
src/plugins/intel_gpu/src/kernel_selector/core/actual_kernels/quantize/quantize_kernel_scale_shift_opt.cpp
+3 -2
View File
@@ -36,8 +36,8 @@ CommonDispatchData QuantizeKernelScaleShift::SetDefault(const quantize_params& p
auto output = params.outputs[0];
if (output.GetLayout() == DataLayout::b_fs_yx_fsv16) {
dispatchData.gws[0] = output.Y().v * output.X().v;
if (output.GetLayout() == DataLayout::b_fs_yx_fsv16 || output.GetLayout() == DataLayout::b_fs_zyx_fsv32) {
dispatchData.gws[0] = output.Z().v *output.Y().v * output.X().v;
dispatchData.gws[1] = Align(output.Feature().v, sub_group_size);
dispatchData.gws[2] = output.Batch().v;
@@ -73,6 +73,7 @@ JitConstants QuantizeKernelScaleShift::GetJitConstants(const quantize_params& pa
JitConstants jit = Parent::GetJitConstants(params, dispatchData);
if (params.outputs[0].GetLayout() == DataLayout::b_fs_yx_fsv16 || params.outputs[0].GetLayout() == DataLayout::bs_fs_yx_bsv32_fsv32 ||
params.outputs[0].GetLayout() == DataLayout::b_fs_zyx_fsv32 ||
params.outputs[0].GetLayout() == DataLayout::bs_fs_yx_bsv16_fsv16 || params.outputs[0].GetLayout() == DataLayout::bs_fs_yx_bsv32_fsv16) {
jit.AddConstant(MakeJitConstant("FEATURE_BLOCKED_FORMAT", true));
jit.AddConstant(MakeJitConstant("GWS_BATCH", 2));