Fix bug

riverlijunjie · Jun 16, 2024 · d7bd900 · d7bd900
1 parent 8a17bea
commit d7bd900
Showing 1 changed file with 2 additions and 14 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
@@ -171,12 +171,6 @@ inline void FUNC(fc_bf_tiled_kernel_default)(
     uint feature_mega_block = gid / (DISPATCH_FSV * DISPATCH_BSV) % (CEIL_DIV(TILE_OUT_F_NUM, TILE_OFM * SIMD) / DISPATCH_FSV);
     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 USE_SLM
     uint out_f = gid * (TILE_OFM * SIMD);
     uint out_b = LWS_BATCHES * TILE_B * (uint)get_group_id(2) + local_id * TILE_B;
@@ -647,13 +641,7 @@ inline void FUNC(fc_bf_tiled_kernel_default)(
                 output_offset += TILE_OUT_B_PITCH;                          \
             } while (false)
 #endif
-        if(left_batch == TILE_B) {
-            CONST_LOOP(TILE_B, WRITE_OUTPUT);
-        } else {
-            unroll_for (uint bi = 0; bi < left_batch; ++bi) {
-                CONST_LOOP(1, WRITE_OUTPUT);
-            }
-        }
+        CONST_LOOP(TILE_B, WRITE_OUTPUT);
         #undef WRITE_OUTPUT
     } else {
         output_offset += sglid;
@@ -679,7 +667,7 @@ inline void FUNC(fc_bf_tiled_kernel_default)(
         //#undef WRITE_OUTPUT
         //#undef WRITE_OUTPUT_FEATURE
 
-        for (uint bi = 0; bi < left_batch; ++bi) {
+        for (uint bi = 0; bi < TILE_B; ++bi) {
             for (uint fi = 0; fi < TILE_OFM; ++fi) {
                 const bool should_write =
 #if IS_DYNAMIC