[GPU] Fix oob in bfyx reorder for BMG

openvinotoolkit · Dec 27, 2024 · 94f6972 · 94f6972
1 parent b25413c
commit 94f6972
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/...gins/intel_gpu/src/kernel_selector/cl_kernels/reorder_data_b_fs_yx_fsv16_fsv32_to_bfyx.cl b/...gins/intel_gpu/src/kernel_selector/cl_kernels/reorder_data_b_fs_yx_fsv16_fsv32_to_bfyx.cl
@@ -66,10 +66,7 @@ KERNEL (reorder_data_b_fs_yx_fsv16_fsv32_to_bfyx)(
 
 
 #if (TILE_SIZE == DEFAULT_TILE_SIZE)
-    // read
-    INPUTVTYPE read_data = AS_INPUTVTYPE(_sub_group_block_read8((const __global uint*)(input) + input_idx_tile));
-
-    // write
+    // write index
     const uint output_idx = OUTPUT_GET_TILED_INDEX(OUTPUT_TILED_ORDER);
 
     if (F_NO_REMAINDER_CONDITION
@@ -79,13 +76,25 @@ KERNEL (reorder_data_b_fs_yx_fsv16_fsv32_to_bfyx)(
     ) {
         #ifdef X_REMAINDER_SIZE
             if (X_REMAINDER_CONDITION) {
+                // read
+                INPUTVTYPE read_data;
+                for (int j = 0; j < X_REMAINDER_SIZE; ++j) {
+                     read_data[j] = input[input_idx_tile+get_sub_group_local_id()+get_max_sub_group_size()*j];
+                }
+                // write
                 for (int i = 0 ; i < X_REMAINDER_SIZE; i++) {
                     output[output_idx + i] = TO_OUTPUT_TYPE(read_data[i]);
                 }
             } else {
+                // read
+                INPUTVTYPE read_data = AS_INPUTVTYPE(_sub_group_block_read8((const __global uint*)(input) + input_idx_tile));
+                // write
                 VSTORE(TO_OUTPUTVTYPE(read_data), 0, output + output_idx);
             }
         #else
+            // read
+            INPUTVTYPE read_data = AS_INPUTVTYPE(_sub_group_block_read8((const __global uint*)(input) + input_idx_tile));
+            // write
             VSTORE(TO_OUTPUTVTYPE(read_data), 0, output + output_idx);
         #endif
     }