Fix WG size

Author: sshlyapn

src/plugins/intel_gpu/src/kernel_selector/cl_kernels/pa_sdpa_ref.cl

@@ -37,7 +37,7 @@
 
 
 REQD_SUB_GROUP_SIZE(SUB_GROUP_SIZE)
-__attribute__((reqd_work_group_size(1, 1, 64)))
+__attribute__((reqd_work_group_size(1, 1, HEAD_SIZE)))
 KERNEL(pa_sdpa_ref)(
     OPTIONAL_SHAPE_INFO_ARG
     __global const INPUT0_TYPE* query,

src/plugins/intel_gpu/src/kernel_selector/kernels/paged_attention/sdpa_kernel_ref.cpp

@@ -9,9 +9,9 @@
 
 namespace kernel_selector {
 
-constexpr size_t HEAD_SIZE = 64;
+constexpr size_t HEAD_SIZE = 128;
 constexpr size_t HEADS_NUM = 32;
-constexpr size_t KV_HEADS_NUM = 4;
+constexpr size_t KV_HEADS_NUM = 2;
 constexpr size_t BLOCK_SIZE = 16;
 constexpr size_t X_BLOCK_SIZE = 8;