Fix ROIAlign kernel error

kelvinchoi-intel · kelvinchoi-intel · commit 3d444d5d9627 · 2024-05-09T00:48:55.000+09:00
diff --git a/src/plugins/intel_gpu/src/kernel_selector/cl_kernels/roi_align_ref.cl b/src/plugins/intel_gpu/src/kernel_selector/cl_kernels/roi_align_ref.cl
@@ -109,7 +109,7 @@ KERNEL(roi_align_ref)(const __global INPUT0_TYPE* src_data,
         }
     }
 #if AVG_POOL
-    pooled_value /= sampling_ratio_x * sampling_ratio_x;
+    pooled_value /= sampling_ratio_x * sampling_ratio_y;
 #endif
 
     dst_data[OUTPUT_GET_INDEX(r, c, y, x)] = ACTIVATION((OUTPUT_TYPE)pooled_value, ACTIVATION_PARAMS);

Original file line number	Diff line number	Diff line change
`@@ -109,7 +109,7 @@ KERNEL(roi_align_ref)(const __global INPUT0_TYPE* src_data,`
`109`	`109`	`}`
`110`	`110`	`}`
`111`	`111`	`#if AVG_POOL`
`112`		`- pooled_value /= sampling_ratio_x * sampling_ratio_x;`
	`112`	`+ pooled_value /= sampling_ratio_x * sampling_ratio_y;`
`113`	`113`	`#endif`
`114`	`114`
`115`	`115`	`dst_data[OUTPUT_GET_INDEX(r, c, y, x)] = ACTIVATION((OUTPUT_TYPE)pooled_value, ACTIVATION_PARAMS);`