Fix bug

openvinotoolkit · Jun 16, 2024 · 9c4ffd1 · 9c4ffd1
1 parent 8a17bea
commit 9c4ffd1
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/plugins/intel_gpu/src/kernel_selector/cl_kernels/fully_connected_gpu_bf_tiled.cl b/src/plugins/intel_gpu/src/kernel_selector/cl_kernels/fully_connected_gpu_bf_tiled.cl
@@ -172,10 +172,12 @@ inline void FUNC(fc_bf_tiled_kernel_default)(
     uint batch_mega_block = gid / (DISPATCH_FSV * DISPATCH_BSV * CEIL_DIV(TILE_OUT_F_NUM, TILE_OFM * SIMD) / DISPATCH_FSV);
 
     uint left_batch = TILE_B;
-    uint last_batch_mega_block = (get_num_groups(0) - 1) / (DISPATCH_FSV * DISPATCH_BSV * CEIL_DIV(TILE_OUT_F_NUM, TILE_OFM * SIMD) / DISPATCH_FSV);
-    uint last_batch_mini_block = (get_num_groups(0) - 1) / DISPATCH_FSV % DISPATCH_BSV;
-    if (batch_mega_block == last_batch_mega_block && batch_mini_block == last_batch_mini_block){
-        left_batch = INPUT0_BATCH_NUM % TILE_B;
+    if((INPUT0_BATCH_NUM * INPUT0_FEATURE_NUM) % TILE_B != 0) {
+        uint last_batch_mega_block = (get_num_groups(0) - 1) / (DISPATCH_FSV * DISPATCH_BSV * CEIL_DIV(TILE_OUT_F_NUM, TILE_OFM * SIMD) / DISPATCH_FSV);
+        uint last_batch_mini_block = (get_num_groups(0) - 1) / DISPATCH_FSV % DISPATCH_BSV;
+        if (batch_mega_block == last_batch_mega_block && batch_mini_block == last_batch_mini_block){
+            left_batch = (INPUT0_BATCH_NUM * INPUT0_FEATURE_NUM) % TILE_B;
+        }
     }
 #if USE_SLM
     uint out_f = gid * (TILE_OFM * SIMD);