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

[GPU] Fix invalid memory access in lrn fsv16 kernel (#14536)

Browse Source
This commit is contained in:
Vladimir Paramuzov 2022-12-13 17:16:13 +04:00 committed by GitHub
parent 9f481e8eaa
commit 78bd280a30
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/plugins/intel_gpu/src/kernel_selector/cl_kernels/lrn_gpu_across_channel_multiple_features_fsv16.cl
View File

@ -29,7 +29,7 @@ KERNEL (lrn_gpu_across_channel_multiple_features_fsv16)(
INPUT0_TYPE res = 0; INPUT0_TYPE res = 0;
for (uint i = 0; i < LOCAL_SIZE; ++i, ++input_offset_f) { for (uint i = 0; i < LOCAL_SIZE; ++i, ++input_offset_f) {
bool non_zero = input_offset_f >= 0 && input_offset_f < INPUT0_FEATURE_NUM; bool non_zero = input_offset_f >= 0 && input_offset_f < INPUT0_FEATURE_NUM;
uint input_idx = INPUT0_GET_INDEX(batch_id, input_offset_f, y, x); uint input_idx = INPUT0_GET_INDEX(batch_id, max(input_offset_f, (int)0), y, x);
val[i] = (int)non_zero * TO_INPUT0_TYPE(ALPHA_VAL_FACTOR_DIV_BY_SIZE) * TO_INPUT0_TYPE(input[input_idx]); val[i] = (int)non_zero * TO_INPUT0_TYPE(ALPHA_VAL_FACTOR_DIV_BY_SIZE) * TO_INPUT0_TYPE(input[input_idx]);
res = mad(val[i], val[i], res); res = mad(val[i], val[i], res);
} }